Update app.py

app.py

@@ -1,13 +1,9 @@
 import os
 import sys
+import logging
 
-# ==========================================
-# 1. 核心补丁：屏蔽 CUDA 和 FlashAttn (防崩)
-# ==========================================
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
-# 欺骗系统：Flash Attention 没安装
-sys.modules["flash_attn"] = None
-
+# --- 1. 基础环境设置 ---
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 import torch
 torch.cuda.is_available = lambda: False
 torch.cuda.device_count = lambda: 0
@@ -15,24 +11,29 @@ def no_op(self, *args, **kwargs): return self
 torch.Tensor.cuda = no_op
 torch.nn.Module.cuda = no_op
 
-print("💉 环境手术完成: CUDA/FlashAttn 已屏蔽。")
+print("💉 CUDA 补丁已注入")
 
-# ==========================================
-# 2. 导入原版逻辑
-# ==========================================
+# --- 2. 导入核心逻辑 ---
 sys.path.append(os.getcwd())
 
 try:
-    # 你的目录里就在根目录下，直接导入！
     import inference_webui as core
     print("✅ 成功导入 inference_webui")
+    
+    # 🛑 关键修正：强制关闭半精度 (CPU 不支持 FP16，也能顺便避开 Flash Attention)
+    if hasattr(core, "is_half"):
+        core.is_half = False
+        print("✅ 强制禁用半精度 (is_half = False)")
+    
+    if hasattr(core, "device"):
+        core.device = "cpu"
+        print("✅ 强制指定设备 (device = cpu)")
+        
 except ImportError:
-    print("❌ 找不到 inference_webui.py，请检查 Files")
+    print("❌ 严重错误：找不到 inference_webui.py")
     sys.exit(1)
 
-# ==========================================
-# 3. 自动加载模型
-# ==========================================
+# --- 3. 自动寻找模型 ---
 def find_real_model(pattern, search_path="."):
     candidates = []
     for root, dirs, files in os.walk(search_path):
@@ -50,27 +51,31 @@ def find_real_model(pattern, search_path="."):
 gpt_path = find_real_model("s1v3.ckpt") or find_real_model("s1bert")
 sovits_path = find_real_model("s2Gv2ProPlus.pth") or find_real_model("s2G")
 
-if gpt_path and sovits_path:
-    try:
-        if hasattr(core, "change_gpt_weights"): core.change_gpt_weights(gpt_path=gpt_path)
-        if hasattr(core, "change_sovits_weights"): core.change_sovits_weights(sovits_path=sovits_path)
-        print(f"🎉 模型加载成功！\nGPT: {gpt_path}\nSoVITS: {sovits_path}")
-    except Exception as e:
-        print(f"⚠️ 模型加载报错: {e}")
-else:
-    print("❌ 未找到模型文件")
+# --- 4. 加载模型 ---
+try:
+    if gpt_path and sovits_path:
+        # 再次确保加载模型时不会启用半精度
+        core.is_half = False 
+        
+        if hasattr(core, "change_gpt_weights"):
+            core.change_gpt_weights(gpt_path=gpt_path)
+        if hasattr(core, "change_sovits_weights"):
+            core.change_sovits_weights(sovits_path=sovits_path)
+        print(f"🎉 模型加载成功！")
+    else:
+        print("❌ 未找到模型文件")
+except Exception as e:
+    print(f"⚠️ 模型加载报错: {e}")
 
-# ==========================================
-# 4. 修复语言参数的推理函数
-# ==========================================
+# --- 5. 推理逻辑 ---
 import soundfile as sf
 import gradio as gr
 import numpy as np
 
 REF_AUDIO = "ref.wav"
 REF_TEXT = "你好"
-# 🔴 关键修正：必须是汉字 "中文"，不能是 "zh"
-LANG_SETTING = "中文" 
+# 🛑 关键修正：语言必须是中文
+REF_LANG = "中文" 
 
 def run_predict(text):
     if not os.path.exists(REF_AUDIO):
@@ -78,20 +83,18 @@ def run_predict(text):
     
     print(f"📥 任务: {text}")
     try:
-        # 自动识别可用函数
+        # 自动识别函数
         inference_func = getattr(core, "get_tts_model", getattr(core, "get_tts_wav", None))
         if not inference_func:
             return None, "❌ 找不到推理函数"
 
         # 核心调用
-        # 这里我们将 prompt_language 和 text_language 都设为 "中文"
-        # 这就是之前 KeyError: 'zh' 的解法
         generator = inference_func(
             ref_wav_path=REF_AUDIO,
             prompt_text=REF_TEXT,
-            prompt_language=LANG_SETTING, # <--- 改成了 "中文"
+            prompt_language=REF_LANG, # 中文
             text=text,
-            text_language=LANG_SETTING,   # <--- 改成了 "中文"
+            text_language="中文",     # 中文
             how_to_cut="凑四句一切",
             top_k=5, top_p=1, temperature=1, ref_free=False
         )
@@ -109,14 +112,12 @@ def run_predict(text):
         traceback.print_exc()
         return None, f"💥 报错: {e}"
 
-# ==========================================
-# 5. 启动界面
-# ==========================================
+# --- 6. 界面 ---
 with gr.Blocks() as app:
-    gr.Markdown(f"### GPT-SoVITS V2 (CPU Worker)")
+    gr.Markdown(f"### GPT-SoVITS V2 (Final CPU)")
     
     with gr.Row():
-        inp = gr.Textbox(label="文本", value="这次肯定没问题了，开始批量生产吧。")
+        inp = gr.Textbox(label="文本", value="这次一定行，不行我就吃键盘。")
         btn = gr.Button("生成")
     
     with gr.Row():