update

modules/models.py

@@ -10,10 +10,7 @@ import gc
 logger = logging.getLogger(__name__)
 
 chat_tts = None
-# 某些平台上，不让在主线程中加载模型，否则会出现错误
-# huggingface Error:
-# RuntimeError: CUDA must not be initialized in the main process on Spaces with Stateless GPU environment.
-# You can look at this Stacktrace to find out which part of your code triggered a CUDA init
+lock = threading.Lock()
 
 
 def load_chat_tts_in_thread():
@@ -40,14 +37,16 @@ def load_chat_tts_in_thread():
 
 
 def initialize_chat_tts():
-    model_thread = threading.Thread(target=load_chat_tts_in_thread)
-    model_thread.start()
-    return model_thread
+    with lock:
+        if chat_tts is None:
+            model_thread = threading.Thread(target=load_chat_tts_in_thread)
+            model_thread.start()
+            model_thread.join()
 
 
 def load_chat_tts():
     if chat_tts is None:
-        initialize_chat_tts().join()
+        initialize_chat_tts()
     if chat_tts is None:
         raise Exception("Failed to load ChatTTS models")
     return chat_tts