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
|