Spaces:

m-ric
/

beam_search_visualizer

Running on Zero

m-ric HF staff commited on Apr 2

Commit

ceb7300

•

1 Parent(s): f9fcae6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -428,7 +428,7 @@ def get_beam_search_html(
     outputs = model.generate(
         **inputs,
         max_new_tokens=number_steps,
-        num_beams=max(number_beams, 2),
         num_return_sequences=num_return_sequences,
         return_dict_in_generate=True,
         length_penalty=length_penalty,

     outputs = model.generate(
         **inputs,
         max_new_tokens=number_steps,
+        num_beams=number_beams,
         num_return_sequences=num_return_sequences,
         return_dict_in_generate=True,
         length_penalty=length_penalty,