Spaces:
Sleeping
Sleeping
Update app.py
Browse filesdouble checking EOS hypothesis
app.py
CHANGED
|
@@ -77,8 +77,7 @@ class ChatBot:
|
|
| 77 |
temperature=0.6,
|
| 78 |
do_sample=True,
|
| 79 |
return_dict_in_generate=True,
|
| 80 |
-
pad_token_id=self.tokenizer.eos_token_id
|
| 81 |
-
eos_token_id="<|im_end|>",
|
| 82 |
)
|
| 83 |
|
| 84 |
# Decode and clean response, with multiple debugs
|
|
|
|
| 77 |
temperature=0.6,
|
| 78 |
do_sample=True,
|
| 79 |
return_dict_in_generate=True,
|
| 80 |
+
pad_token_id=self.tokenizer.eos_token_id
|
|
|
|
| 81 |
)
|
| 82 |
|
| 83 |
# Decode and clean response, with multiple debugs
|