Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -33,7 +33,7 @@ def generate(prompt, history):
|
|
| 33 |
|
| 34 |
with torch.no_grad():
|
| 35 |
outputs = model.generate(
|
| 36 |
-
inputs,
|
| 37 |
max_new_tokens=512,
|
| 38 |
)
|
| 39 |
|
|
|
|
| 33 |
|
| 34 |
with torch.no_grad():
|
| 35 |
outputs = model.generate(
|
| 36 |
+
**inputs,
|
| 37 |
max_new_tokens=512,
|
| 38 |
)
|
| 39 |
|