Spaces:

m-ric
/

beam_search_visualizer

Running on Zero

m-ric HF staff commited on Apr 2

Commit

918dc40

•

1 Parent(s): 9ff548f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -429,7 +429,7 @@ def get_beam_search_html(
         **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,
         output_scores=True,

         **inputs,
         max_new_tokens=number_steps,
         num_beams=number_beams,
+        num_return_sequences=min(num_return_sequences, number_beams),
         return_dict_in_generate=True,
         length_penalty=length_penalty,
         output_scores=True,