lvwerra HF staff commited on
Commit
c444169
1 Parent(s): b98f2e5

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -317,4 +317,4 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
317
  if tokens[-1] == "<|eot_id|>":
318
  break
319
 
320
- yield update_notebook_display(notebook_data), messages
 
317
  if tokens[-1] == "<|eot_id|>":
318
  break
319
 
320
+ yield update_notebook_display(notebook_data), notebook_data, messages