semantic-entropy-probes

Sleeping

s-a-malik commited on Jul 16, 2024

Commit

cdf250a

1 Parent(s): 2c726a9

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -336,8 +336,8 @@ def generate(
         # print(probe_pred.shape, probe_pred)
         # decode one token at a time
         output_id = outputs.sequences[0, input_ids.shape[1]+i]
-        print(output_id, output_word, probe_pred)
         output_word = tokenizer.decode(output_id)
         new_highlighted_text = highlight_text(output_word, probe_pred)
         highlighted_text += new_highlighted_text

         # print(probe_pred.shape, probe_pred)
         # decode one token at a time
         output_id = outputs.sequences[0, input_ids.shape[1]+i]
         output_word = tokenizer.decode(output_id)
+        print(output_id, output_word, probe_pred)
         new_highlighted_text = highlight_text(output_word, probe_pred)
         highlighted_text += new_highlighted_text