update

modules/models.py

@@ -10,15 +10,20 @@ 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
 load_event = threading.Event()
 
 
 def load_chat_tts_in_thread():
     global chat_tts
     if chat_tts:
-        load_event.set()  # 如果已经加载过，直接设置事件
+        load_event.set()
         return
 
+    logger.info("Loading ChatTTS models")
     chat_tts = ChatTTS.Chat()
     chat_tts.load_models(
         compile=config.runtime_env_vars.compile,
@@ -33,17 +38,19 @@ def load_chat_tts_in_thread():
     )
 
     devices.torch_gc()
-    load_event.set()  # 设置事件，表示加载完成
+    load_event.set()
+    logger.info("ChatTTS models loaded")
 
 
 def initialize_chat_tts():
     model_thread = threading.Thread(target=load_chat_tts_in_thread)
     model_thread.start()
+    return model_thread
 
 
 def load_chat_tts():
     if chat_tts is None:
-        initialize_chat_tts()
+        initialize_chat_tts().join()
     load_event.wait()
     return chat_tts