修复本地模型在windows上的兼容性

request_llms/bridge_chatglm.py

@@ -4,14 +4,13 @@
 
 from transformers import AutoModel, AutoTokenizer
 from toolbox import get_conf, ProxyNetworkActivate
-from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns, SingletonLocalLLM
+from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns
 
 
 
 # ------------------------------------------------------------------------------------------------------------------------
 # 🔌💻 Local Model
 # ------------------------------------------------------------------------------------------------------------------------
-@SingletonLocalLLM
 class GetGLM2Handle(LocalLLMHandle):
 
     def load_model_info(self):

request_llms/bridge_chatglm3.py

@@ -4,14 +4,13 @@
 
 from transformers import AutoModel, AutoTokenizer
 from toolbox import get_conf, ProxyNetworkActivate
-from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns, SingletonLocalLLM
+from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns
 
 
 
 # ------------------------------------------------------------------------------------------------------------------------
 # 🔌💻 Local Model
 # ------------------------------------------------------------------------------------------------------------------------
-@SingletonLocalLLM
 class GetGLM3Handle(LocalLLMHandle):
 
     def load_model_info(self):

request_llms/bridge_chatglmonnx.py

@@ -8,7 +8,7 @@
 import importlib
 from toolbox import update_ui, get_conf
 from multiprocessing import Process, Pipe
-from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns, SingletonLocalLLM
+from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns
 
 from .chatglmoonx import ChatGLMModel, chat_template
 
@@ -17,7 +17,6 @@
 # ------------------------------------------------------------------------------------------------------------------------
 # 🔌💻 Local Model
 # ------------------------------------------------------------------------------------------------------------------------
-@SingletonLocalLLM
 class GetONNXGLMHandle(LocalLLMHandle):
 
     def load_model_info(self):

request_llms/bridge_internlm.py

@@ -7,7 +7,7 @@
 import importlib
 from toolbox import update_ui, get_conf
 from multiprocessing import Process, Pipe
-from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns, SingletonLocalLLM
+from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns
 
 
 # ------------------------------------------------------------------------------------------------------------------------
@@ -34,7 +34,6 @@ def combine_history(prompt, hist):
 # ------------------------------------------------------------------------------------------------------------------------
 # 🔌💻 Local Model
 # ------------------------------------------------------------------------------------------------------------------------
-@SingletonLocalLLM
 class GetInternlmHandle(LocalLLMHandle):
 
     def load_model_info(self):

request_llms/bridge_llama2.py

@@ -5,14 +5,13 @@
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from toolbox import update_ui, get_conf, ProxyNetworkActivate
 from multiprocessing import Process, Pipe
-from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns, SingletonLocalLLM
+from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns
 from threading import Thread
 
 
 # ------------------------------------------------------------------------------------------------------------------------
 # 🔌💻 Local Model
 # ------------------------------------------------------------------------------------------------------------------------
-@SingletonLocalLLM
 class GetONNXGLMHandle(LocalLLMHandle):
 
     def load_model_info(self):

request_llms/bridge_qwen.py

@@ -8,14 +8,13 @@
 import importlib
 from toolbox import update_ui, get_conf
 from multiprocessing import Process, Pipe
-from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns, SingletonLocalLLM
+from .local_llm_class import LocalLLMHandle, get_local_llm_predict_fns
 
 
 
 # ------------------------------------------------------------------------------------------------------------------------
 # 🔌💻 Local Model
 # ------------------------------------------------------------------------------------------------------------------------
-@SingletonLocalLLM
 class GetONNXGLMHandle(LocalLLMHandle):
 
     def load_model_info(self):

request_llms/local_llm_class.py

@@ -76,7 +76,6 @@ def __init__(self):
         self.parent_state, self.child_state = create_queue_pipe()
         # allow redirect_stdout
         self.std_tag = "[Subprocess Message] "
-        self.child.write = lambda x: self.child.send(self.std_tag + x)
         self.running = True
         self._model = None
         self._tokenizer = None
@@ -137,6 +136,8 @@ def check_dependency(self):
     def run(self):
         # 🏃‍♂️🏃‍♂️🏃‍♂️ run in child process
         # 第一次运行，加载参数
+        self.child.flush = lambda *args: None
+        self.child.write = lambda x: self.child.send(self.std_tag + x)
         reset_tqdm_output()
         self.set_state("`尝试加载模型`")
         try:
@@ -220,7 +221,7 @@ def predict_no_ui_long_connection(inputs, llm_kwargs, history=[], sys_prompt="",
         """
             refer to request_llms/bridge_all.py
         """
-        _llm_handle = LLMSingletonClass()
+        _llm_handle = SingletonLocalLLM(LLMSingletonClass)()
         if len(observe_window) >= 1:
             observe_window[0] = load_message + "\n\n" + _llm_handle.get_state()
         if not _llm_handle.running:
@@ -268,7 +269,7 @@ def predict(inputs, llm_kwargs, plugin_kwargs, chatbot, history=[], system_promp
         """
         chatbot.append((inputs, ""))
 
-        _llm_handle = LLMSingletonClass()
+        _llm_handle = SingletonLocalLLM(LLMSingletonClass)()
         chatbot[-1] = (inputs, load_message + "\n\n" + _llm_handle.get_state())
         yield from update_ui(chatbot=chatbot, history=[])
         if not _llm_handle.running: