isakzhang commited on
Commit
6e1fb38
1 Parent(s): 6d57eb4

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 = []