Update app.py
Browse files
app.py
CHANGED
|
@@ -57,6 +57,8 @@ def predict(message, history):
|
|
| 57 |
top_k= 60,
|
| 58 |
temperature=0.2,
|
| 59 |
num_beams=1,
|
|
|
|
|
|
|
| 60 |
stopping_criteria=StoppingCriteriaList([stop]),
|
| 61 |
repetition_penalty=1.1,
|
| 62 |
)
|
|
|
|
| 57 |
top_k= 60,
|
| 58 |
temperature=0.2,
|
| 59 |
num_beams=1,
|
| 60 |
+
eos_token_id=tokenizer.eos_token_id,
|
| 61 |
+
pad_token_id=tokenizer.pad_token_id,
|
| 62 |
stopping_criteria=StoppingCriteriaList([stop]),
|
| 63 |
repetition_penalty=1.1,
|
| 64 |
)
|