nxphi47 commited on
Commit
764cf91
1 Parent(s): 55ebaa5

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: