bug fix

ZurichRain · Jan 9, 2023 · 5c73bcc · 5c73bcc
1 parent eaac3f7
commit 5c73bcc
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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()