Spaces:
Running
Running
Update utils.py
Browse files
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
|