Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ def generate(
|
|
| 61 |
top_p=top_p,
|
| 62 |
top_k=top_k,
|
| 63 |
temperature=temperature,
|
| 64 |
-
num_beams=
|
| 65 |
repetition_penalty=repetition_penalty,
|
| 66 |
)
|
| 67 |
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
|
|
|
| 61 |
top_p=top_p,
|
| 62 |
top_k=top_k,
|
| 63 |
temperature=temperature,
|
| 64 |
+
num_beams=1,
|
| 65 |
repetition_penalty=repetition_penalty,
|
| 66 |
)
|
| 67 |
t = Thread(target=model.generate, kwargs=generate_kwargs)
|