Spaces:
Running
on
Zero
Running
on
Zero
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 = []
|