Update chat_final.py

chat_final.py

@@ -1,4 +1,3 @@
-print('Loading...')
 from src.model_run import RWKV_RNN
 import numpy as np
 import os, copy, types, gc, sys
@@ -32,7 +31,6 @@ interface = ":"
 os.environ["RWKV_RUN_DEVICE"] = args.RUN_DEVICE
 MODEL_NAME = args.MODEL_NAME
 
-print(f'loading... {MODEL_NAME}')
 model = RWKV_RNN(args)
 
 model_tokens = []
@@ -64,7 +62,6 @@ def load_all_stat(name):
     model_tokens = copy.deepcopy(all_state[name]['token'])
     return all_state[name]['out']
 
-print(f'\nRun prompt...')
 
 out = ""
 gc.collect()
@@ -72,9 +69,6 @@ gc.collect()
 save_all_stat('chat_init', out)
 save_all_stat('chat', out)  # ensure that 'chat' key is added to all_state
 
-print(f'### prompt ###\n[{tokenizer.tokenizer.decode(model_tokens)}]\n')
-
-
 def reply_msg_generator():
     while True:
         msg = yield
@@ -125,7 +119,7 @@ def on_message_generator():
         yield '\n'  # Yield a newline at the end of the response
         message = yield  # Get the next message
 
-print('Start chatting with Daniel!')
+print('Start chatting with Daniel! Pretend to pick up the phone.')
 
 on_message_gen = on_message_generator()
 next_message = on_message_gen.__next__()  # Start the generator