Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -37,6 +37,7 @@ def respond(
|
|
| 37 |
do_sample=True,
|
| 38 |
temperature=temperature,
|
| 39 |
top_p=top_p,
|
|
|
|
| 40 |
)
|
| 41 |
|
| 42 |
thread = threading.Thread(target=model.generate, kwargs=generate_kwargs)
|
|
|
|
| 37 |
do_sample=True,
|
| 38 |
temperature=temperature,
|
| 39 |
top_p=top_p,
|
| 40 |
+
eos_token_id=tokenizer.eos_token_id,
|
| 41 |
)
|
| 42 |
|
| 43 |
thread = threading.Thread(target=model.generate, kwargs=generate_kwargs)
|