Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -116,7 +116,7 @@ def generate_response(user_input):
|
|
116 |
with torch.no_grad():
|
117 |
outputs = proc_model.generate(
|
118 |
inputs,
|
119 |
-
max_new_tokens=
|
120 |
)
|
121 |
|
122 |
response = proc_tokenizer.batch_decode(
|
|
|
116 |
with torch.no_grad():
|
117 |
outputs = proc_model.generate(
|
118 |
inputs,
|
119 |
+
max_new_tokens=256,
|
120 |
)
|
121 |
|
122 |
response = proc_tokenizer.batch_decode(
|