decoding_visualizer

Running on T4

m-ric commited on Apr 2, 2024

Commit

bf09de3

verified ·

1 Parent(s): 9f940a0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -448,7 +448,6 @@ def get_beam_search_html(
             markdown += f"\n- Score `{outputs.sequences_scores[i]:.2f}`: `{clean(sequence.replace('<s> ', ''))}`"
     else:
         markdown += f"\n- `{clean(decoded_sequences[0].replace('<s> ', ''))}`"
-    print(outputs.logits)
     if number_beams > 1:
         original_tree = generate_beams(

             markdown += f"\n- Score `{outputs.sequences_scores[i]:.2f}`: `{clean(sequence.replace('<s> ', ''))}`"
     else:
         markdown += f"\n- `{clean(decoded_sequences[0].replace('<s> ', ''))}`"
     if number_beams > 1:
         original_tree = generate_beams(