lvwerra HF staff commited on
Commit
c2726bd
·
verified ·
1 Parent(s): ef8e02d

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +7 -7
utils.py CHANGED
@@ -210,15 +210,15 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
210
  print("Start!")
211
  notebook_data, code_cell_counter = create_base_notebook(messages)
212
  try:
213
- input_tokens = tokenizer.apply_chat_template(
214
- messages,
215
- builtin_tools=["code_interpreter"],
216
- add_generation_prompt=True
217
- )
218
- model_input = tokenizer.decode(input_tokens)
219
-
220
  #code_cell_counter = 0
221
  while True:
 
 
 
 
 
 
 
222
  response_stream = client.text_generation(
223
  model=model,
224
  prompt=model_input,
 
210
  print("Start!")
211
  notebook_data, code_cell_counter = create_base_notebook(messages)
212
  try:
 
 
 
 
 
 
 
213
  #code_cell_counter = 0
214
  while True:
215
+ input_tokens = tokenizer.apply_chat_template(
216
+ messages,
217
+ builtin_tools=["code_interpreter"],
218
+ add_generation_prompt=True
219
+ )
220
+ model_input = tokenizer.decode(input_tokens)
221
+
222
  response_stream = client.text_generation(
223
  model=model,
224
  prompt=model_input,