Update multipurpose_chatbot/engines/transformers_engine.py

multipurpose_chatbot/engines/transformers_engine.py

@@ -32,6 +32,7 @@ from .base_engine import BaseEngine
 
 # ! Remember to use static cache
 
+
 from transformers import (
     GenerationConfig,
     GenerationMixin,
@@ -561,6 +562,7 @@ class TransformersEngine(BaseEngine):
                 temperature=temperature, 
                 max_new_tokens=max_tokens, 
                 pad_token_id=self.tokenizer.pad_token_id,
+                eos_token_id=self.tokenizer.eos_token_id,
             )
 
             out_tokens = []