Update services/chat_service.py
Browse files- services/chat_service.py +1 -1
services/chat_service.py
CHANGED
|
@@ -247,7 +247,7 @@ class ChatService:
|
|
| 247 |
do_sample=True,
|
| 248 |
no_repeat_ngram_size=3,
|
| 249 |
early_stopping=False, # True num_beams=3, # Increase number of beams if beam search is needed
|
| 250 |
-
#pad_token_id=self.tokenizer.
|
| 251 |
)
|
| 252 |
|
| 253 |
response = self.tokenizer.decode(
|
|
|
|
| 247 |
do_sample=True,
|
| 248 |
no_repeat_ngram_size=3,
|
| 249 |
early_stopping=False, # True num_beams=3, # Increase number of beams if beam search is needed
|
| 250 |
+
#pad_token_id=self.tokenizer.eos_token_id, #eos_token_id
|
| 251 |
)
|
| 252 |
|
| 253 |
response = self.tokenizer.decode(
|