Spaces:
Running
Running
Update utils.py
Browse files
utils.py
CHANGED
@@ -223,7 +223,7 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
|
|
223 |
prompt=model_input,
|
224 |
details=True,
|
225 |
stream=True,
|
226 |
-
|
227 |
)
|
228 |
|
229 |
assistant_response = ""
|
|
|
223 |
prompt=model_input,
|
224 |
details=True,
|
225 |
stream=True,
|
226 |
+
max_new_tokens=max_new_tokens,
|
227 |
)
|
228 |
|
229 |
assistant_response = ""
|