bug fix

3 years ago · 5c73bccd5a
parent eaac3f7e66
commit 5c73bccd5a
1 changed files with 2 additions and 0 deletions
--- a/RWKV-v4neo/chat.py
+++ b/RWKV-v4neo/chat.py
@ -140,6 +140,8 @@ def reply_msg(msg):
    print('Bot:', msg + '\n')
 def on_message(message):
    global model_tokens, current_state
    srv = 'dummy_server'
    msg = message.strip()