Spaces:
Running
on
Zero
Running
on
Zero
Update multipurpose_chatbot/engines/transformers_engine.py
Browse files
multipurpose_chatbot/engines/transformers_engine.py
CHANGED
@@ -453,8 +453,8 @@ class TransformersEngine(BaseEngine):
|
|
453 |
if "<|im_start|>assistant\n" in response:
|
454 |
response = response.split("<|im_start|>assistant\n")[-1]
|
455 |
num_tokens += 1
|
456 |
-
print(f"{response}", end='\r')
|
457 |
-
sys.stdout.flush()
|
458 |
yield response, num_tokens
|
459 |
|
460 |
if response is not None:
|
|
|
453 |
if "<|im_start|>assistant\n" in response:
|
454 |
response = response.split("<|im_start|>assistant\n")[-1]
|
455 |
num_tokens += 1
|
456 |
+
# print(f"{response}", end='\r')
|
457 |
+
# sys.stdout.flush()
|
458 |
yield response, num_tokens
|
459 |
|
460 |
if response is not None:
|