nxphi47 commited on
Commit
12f24ba
1 Parent(s): 3d983d5

Update multipurpose_chatbot/engines/transformers_engine.py

Browse files
multipurpose_chatbot/engines/transformers_engine.py CHANGED
@@ -567,7 +567,7 @@ class TransformersEngine(BaseEngine):
567
  response = None
568
  for index, token in enumerate(generator):
569
  out_tokens.extend(token.tolist())
570
- response = self.tokenizer.decode(out_tokens)
571
  if "<|im_start|>assistant\n" in response:
572
  response = response.split("<|im_start|>assistant\n")[-1]
573
  num_tokens += 1
 
567
  response = None
568
  for index, token in enumerate(generator):
569
  out_tokens.extend(token.tolist())
570
+ response = self.tokenizer.decode(out_tokens, skip_special_tokens=True)
571
  if "<|im_start|>assistant\n" in response:
572
  response = response.split("<|im_start|>assistant\n")[-1]
573
  num_tokens += 1