From 5c73bccd5a2eee8f420dd9ae9c4818661191a888 Mon Sep 17 00:00:00 2001
From: BlinkDL <a@a.com>
Date: Mon, 9 Jan 2023 19:40:08 +0000
Subject: [PATCH] bug fix

---
 RWKV-v4neo/chat.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/RWKV-v4neo/chat.py b/RWKV-v4neo/chat.py
index 4e95bd7..cb4a49e 100644
--- 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()