Update chat.py

hiyouga · Jul 18, 2023 · d4d6250 · d4d6250
1 parent 162cc53
commit d4d6250
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/glmtuner/webui/chat.py b/src/glmtuner/webui/chat.py
@@ -11,10 +11,12 @@
 
 class WebChatModel(ChatModel):
 
-    def __init__(self):
+    def __init__(self, *args):
         self.model = None
         self.tokenizer = None
         self.generating_args = GeneratingArguments()
+        if len(args) != 0:
+            super().__init__(*args)
 
     def load_model(
         self,