更新

requirements.txt

@@ -1,10 +1,10 @@
 # PyTorch and related libraries
-torch
+torch>=2.0.1
 torchvision
 torchaudio
 
 # Hugging Face transformers library
-transformers
+transformers==4.38.2
 
 # Configuration management with OmegaConf
 omegaconf
@@ -13,7 +13,7 @@ omegaconf
 ipywidgets
 
 # Gradio for creating web UIs
-gradio
+gradio>=5.16.0
 
 # Vector quantization for PyTorch
 vector_quantize_pytorch

web/app_gpu.py

@@ -36,7 +36,7 @@ def init_chat(args):
     logger.info("loading Chat2TTS model..., start source:" + source)
 
     device=select_device()
-    logger.info("loading ChatTTS device :" + str(device))
+    logger.info("----------loading ChatTTS device :" + str(device))
 
     if chat.load_models(source=source, local_path="D:\\chenjgspace\\ai-model\\chattts"):
         print("Models loaded successfully.")