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)
|