lvwerra HF staff commited on
Commit
6d80856
·
verified ·
1 Parent(s): 6b6ff22

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -223,7 +223,7 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
223
  prompt=model_input,
224
  details=True,
225
  stream=True,
226
- max_tokens=max_new_tokens,
227
  )
228
 
229
  assistant_response = ""
 
223
  prompt=model_input,
224
  details=True,
225
  stream=True,
226
+ max_new_tokens=max_new_tokens,
227
  )
228
 
229
  assistant_response = ""