Spaces:
Runtime error
Runtime error
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
|