beam field fix

demo_watermark.py

@@ -97,7 +97,7 @@ def parse_args():
     parser.add_argument(
         "--n_beams",
         type=int,
-        choices=[1,4,8],
+        default=1,
         help="Number of beams to use for beam search. 1 is normal greedy decoding",
     )
     parser.add_argument(
@@ -399,7 +399,7 @@ def run_gradio(args, model=None, device=None, tokenizer=None):
                 return gr.update(visible=False)
             elif value == "greedy":
                 return gr.update(visible=True)
-        def update_n_beams(session_state, value): session_state.n_beams = int(value); return session_state
+        def update_n_beams(session_state, value): session_state.n_beams = value; return session_state
         def update_max_new_tokens(session_state, value): session_state.max_new_tokens = int(value); return session_state
         def update_ignore_repeated_bigrams(session_state, value): session_state.ignore_repeated_bigrams = value; return session_state
         def update_normalizers(session_state, value): session_state.normalizers = value; return session_state