fix: 优化控制台闪烁的问题

cli_demo.py

@@ -39,20 +39,12 @@ def main():
  os.system(clear_command)
  print("欢迎使用 ChatGLM-6B 模型，输入内容即可进行对话，clear 清空对话历史，stop 终止程序")
  continue
- count = 0
+ prev_response = ""
+ print("ChatGLM-6B：", end="", flush=True)
  for response, history in model.stream_chat(tokenizer, query, history=history):
- if stop_stream:
- stop_stream = False
- break
- else:
- count += 1
- if count % 8 == 0:
- os.system(clear_command)
- print(build_prompt(history), flush=True)
- signal.signal(signal.SIGINT, signal_handler)
- os.system(clear_command)
- print(build_prompt(history), flush=True)
-
+ print(response[len(prev_response):], end="", flush=True)
+ prev_response = response
+ print("\n", end="", flush=True)
 
 if __name__ == "__main__":
  main()