Spaces:
Running
Running
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -229,6 +229,8 @@ def run_interactive_notebook(client, model, tokenizer, messages, sbx, max_new_to
|
|
| 229 |
add_generation_prompt=True
|
| 230 |
)
|
| 231 |
model_input = tokenizer.decode(input_tokens)
|
|
|
|
|
|
|
| 232 |
|
| 233 |
response_stream = client.text_generation(
|
| 234 |
model=model,
|
|
|
|
| 229 |
add_generation_prompt=True
|
| 230 |
)
|
| 231 |
model_input = tokenizer.decode(input_tokens)
|
| 232 |
+
|
| 233 |
+
print(f"Model input:\n{model_input}\n{'='*80}")
|
| 234 |
|
| 235 |
response_stream = client.text_generation(
|
| 236 |
model=model,
|