lvwerra HF staff commited on
Commit
fa34d67
·
verified ·
1 Parent(s): 4c74a4e

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -218,7 +218,7 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
218
 
219
  #code_cell_counter = 0
220
  while True:
221
- response_stream = client.chat.completions.create(
222
  model=model,
223
  messages=model_input,
224
  details=True,
 
218
 
219
  #code_cell_counter = 0
220
  while True:
221
+ response_stream = client.text_generation(
222
  model=model,
223
  messages=model_input,
224
  details=True,