Spaces:
Paused
Paused
Update multipurpose_chatbot/engines/transformers_engine.py
Browse files
multipurpose_chatbot/engines/transformers_engine.py
CHANGED
|
@@ -579,6 +579,7 @@ class TransformersEngine(BaseEngine):
|
|
| 579 |
temperature=temperature,
|
| 580 |
max_new_tokens=max_tokens,
|
| 581 |
pad_token_id=self.tokenizer.pad_token_id,
|
|
|
|
| 582 |
)
|
| 583 |
|
| 584 |
out_tokens = []
|
|
|
|
| 579 |
temperature=temperature,
|
| 580 |
max_new_tokens=max_tokens,
|
| 581 |
pad_token_id=self.tokenizer.pad_token_id,
|
| 582 |
+
eos_token_id=self.tokenizer.eos_token_id,
|
| 583 |
)
|
| 584 |
|
| 585 |
out_tokens = []
|