Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -92,7 +92,7 @@ def generate(
|
|
| 92 |
num_beams=1,
|
| 93 |
repetition_penalty=repetition_penalty,
|
| 94 |
)
|
| 95 |
-
t = Thread(target=model.generate)
|
| 96 |
t.start()
|
| 97 |
|
| 98 |
outputs = []
|
|
|
|
| 92 |
num_beams=1,
|
| 93 |
repetition_penalty=repetition_penalty,
|
| 94 |
)
|
| 95 |
+
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
| 96 |
t.start()
|
| 97 |
|
| 98 |
outputs = []
|