Spaces:

m-ric
/

beam_search_visualizer

Running on Zero

m-ric HF staff commited on Mar 25

Commit

5e72e33

•

1 Parent(s): 1447899

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -368,7 +368,7 @@ def get_beam_search_html(input_text, number_steps, number_beams, length_penalty)
         num_beams=number_beams,
         num_return_sequences=number_beams,
         return_dict_in_generate=True,
-        length_penalty=-10.0,
         output_scores=True,
         do_sample=False,
     )

         num_beams=number_beams,
         num_return_sequences=number_beams,
         return_dict_in_generate=True,
+        length_penalty=length_penalty,
         output_scores=True,
         do_sample=False,
     )