Spaces:
Paused
Paused
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:
|