decoding_visualizer

Running on T4

Jofthomas commited on Jan 23

Commit

885e92b

verified ·

1 Parent(s): 7d8f9c5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -424,7 +424,7 @@ def get_beam_search_html(
     input_text,
     number_steps=12,
     number_beams=1,
-    length_penalty=3.0,
     num_return_sequences=3
 ):
     inputs = tokenizer([input_text], return_tensors="pt")

     input_text,
     number_steps=12,
     number_beams=1,
+    length_penalty=-3.0,
     num_return_sequences=3
 ):
     inputs = tokenizer([input_text], return_tensors="pt")