lvwerra HF staff commited on
Commit
650cd70
·
verified ·
1 Parent(s): 3ab9656

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +0 -2
utils.py CHANGED
@@ -316,7 +316,5 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
316
  messages.append({"role": "assistant", "content": assistant_response})
317
  if tokens[-1] == "<|eot_id|>":
318
  break
319
- finally:
320
- sbx.kill()
321
 
322
  yield update_notebook_display(notebook_data), messages
 
316
  messages.append({"role": "assistant", "content": assistant_response})
317
  if tokens[-1] == "<|eot_id|>":
318
  break
 
 
319
 
320
  yield update_notebook_display(notebook_data), messages