debug

app.py

@@ -9,6 +9,19 @@ import shutil
 from huggingface_hub import snapshot_download, hf_hub_download
 import requests
 import subprocess
+import json
+
+# 获取当前工作目录的绝对路径
+BASE_DIR = os.path.abspath(os.getcwd())
+
+# 修复相对路径为绝对路径的函数
+def fix_path(path):
+    if path is None:
+        return None
+    # 如果是相对路径（以./开头），转换为绝对路径
+    if isinstance(path, str) and path.startswith('./'):
+        return os.path.join(BASE_DIR, path[2:])
+    return path
 
 # 检查并安装必要的依赖
 def install_dependencies():
@@ -100,6 +113,33 @@ def download_amphion_code():
             if not os.path.exists(local_path):
                 with open(local_path, "w") as f:
                     f.write("# Placeholder file\n")
+    
+    # 下载特殊文件：hubert_large_l18_mean_std.npz
+    try:
+        # 确保目录存在
+        os.makedirs(os.path.join(os.getcwd(), "models/vc/vevo/config"), exist_ok=True)
+        
+        # 尝试从HuggingFace下载
+        try:
+            hf_hub_download(
+                repo_id=REPO_ID,
+                filename="tokenizer/vq8192/hubert_large_l18_mean_std.npz",
+                cache_dir=CACHE_DIR,
+                local_dir=os.path.join(os.getcwd(), "models/vc/vevo/config"),
+                local_dir_use_symlinks=False
+            )
+            print("成功下载: hubert_large_l18_mean_std.npz")
+        except Exception as e:
+            print(f"无法从HuggingFace下载hubert_large_l18_mean_std.npz: {str(e)}")
+            # 尝试从GitHub下载
+            hubert_url = "https://huggingface.co/amphion/Vevo/resolve/main/tokenizer/vq8192/hubert_large_l18_mean_std.npz"
+            response = requests.get(hubert_url)
+            if response.status_code == 200:
+                with open(os.path.join(os.getcwd(), "models/vc/vevo/config/hubert_large_l18_mean_std.npz"), "wb") as f:
+                    f.write(response.content)
+                print("成功从HuggingFace下载: hubert_large_l18_mean_std.npz")
+    except Exception as e:
+        print(f"下载hubert_large_l18_mean_std.npz时出错: {str(e)}")
 
 # 先下载必要的代码文件
 download_amphion_code()
@@ -159,10 +199,10 @@ class VevoGradioApp:
         self.pipelines = {}
         # 配置文件路径
         self.config_paths = {
-            "vq32tovq8192": "./models/vc/vevo/config/Vq32ToVq8192.json",
-            "vq8192tomels": "./models/vc/vevo/config/Vq8192ToMels.json",
-            "phonetovq8192": "./models/vc/vevo/config/PhoneToVq8192.json",
-            "vocoder": "./models/vc/vevo/config/Vocoder.json"
+            "vq32tovq8192": fix_path("./models/vc/vevo/config/Vq32ToVq8192.json"),
+            "vq8192tomels": fix_path("./models/vc/vevo/config/Vq8192ToMels.json"),
+            "phonetovq8192": fix_path("./models/vc/vevo/config/PhoneToVq8192.json"),
+            "vocoder": fix_path("./models/vc/vevo/config/Vocoder.json")
         }
         
         # 确保配置文件存在
@@ -170,7 +210,8 @@ class VevoGradioApp:
         
     def download_configs(self):
         """下载必要的配置文件"""
-        os.makedirs("./models/vc/vevo/config", exist_ok=True)
+        config_dir = fix_path("./models/vc/vevo/config")
+        os.makedirs(config_dir, exist_ok=True)
         config_files = {
             "Vq32ToVq8192.json": "https://raw.githubusercontent.com/open-mmlab/Amphion/main/models/vc/vevo/config/Vq32ToVq8192.json",
             "Vq8192ToMels.json": "https://raw.githubusercontent.com/open-mmlab/Amphion/main/models/vc/vevo/config/Vq8192ToMels.json",
@@ -179,7 +220,7 @@ class VevoGradioApp:
         }
         
         for filename, url in config_files.items():
-            target_path = f"./models/vc/vevo/config/{filename}"
+            target_path = os.path.join(config_dir, filename)
             if not os.path.exists(target_path):
                 try:
                     response = requests.get(url)
@@ -197,6 +238,31 @@ class VevoGradioApp:
                     with open(target_path, 'w') as f:
                         f.write('{}')
                     print(f"无法下载配置文件 {filename}，已创建占位符")
+                    
+        # 修改配置文件中的相对路径
+        self.update_config_paths()
+    
+    def update_config_paths(self):
+        """修改配置文件中的相对路径"""
+        try:
+            # 修改Vq8192ToMels.json文件中的路径引用
+            vq8192_path = self.config_paths["vq8192tomels"]
+            if os.path.exists(vq8192_path):
+                with open(vq8192_path, 'r') as f:
+                    config = json.load(f)
+                
+                # 修改hubert_large_l18_mean_std.npz路径
+                if "model" in config and "representation_stat_mean_var_path" in config["model"]:
+                    # 替换为绝对路径
+                    hubert_stat_path = fix_path("./models/vc/vevo/config/hubert_large_l18_mean_std.npz")
+                    config["model"]["representation_stat_mean_var_path"] = hubert_stat_path
+                    
+                    # 保存修改后的配置
+                    with open(vq8192_path, 'w') as f:
+                        json.dump(config, f, indent=2)
+                    print("成功更新配置文件中的路径")
+        except Exception as e:
+            print(f"更新配置文件路径时出错: {str(e)}")
     
     def init_voice_conversion_pipeline(self):
         """初始化语音转换管道"""
@@ -264,7 +330,16 @@ class VevoGradioApp:
             except Exception as e:
                 print(f"初始化语音转换管道时出错: {str(e)}")
                 # 创建一个占位符管道
-                self.pipelines["voice"] = VevoInferencePipeline(device=self.device)
+                try:
+                    # 尝试提供必要的配置文件
+                    self.pipelines["voice"] = VevoInferencePipeline(
+                        device=self.device,
+                        fmt_cfg_path=self.config_paths["vq8192tomels"],
+                        vocoder_cfg_path=self.config_paths["vocoder"],
+                    )
+                except:
+                    # 如果还是失败，创建最简单的管道
+                    self.pipelines["voice"] = VevoInferencePipeline(device=self.device)
             
         return self.pipelines["voice"]
     
@@ -311,7 +386,16 @@ class VevoGradioApp:
             except Exception as e:
                 print(f"初始化音色转换管道时出错: {str(e)}")
                 # 创建一个占位符管道
-                self.pipelines["timbre"] = VevoInferencePipeline(device=self.device)
+                try:
+                    # 尝试提供必要的配置文件
+                    self.pipelines["timbre"] = VevoInferencePipeline(
+                        device=self.device,
+                        fmt_cfg_path=self.config_paths["vq8192tomels"],
+                        vocoder_cfg_path=self.config_paths["vocoder"],
+                    )
+                except:
+                    # 如果还是失败，创建最简单的管道
+                    self.pipelines["timbre"] = VevoInferencePipeline(device=self.device)
             
         return self.pipelines["timbre"]
     
@@ -369,7 +453,17 @@ class VevoGradioApp:
             except Exception as e:
                 print(f"初始化TTS管道时出错: {str(e)}")
                 # 创建一个占位符管道
-                self.pipelines["tts"] = VevoInferencePipeline(device=self.device)
+                try:
+                    # 尝试提供必要的配置文件
+                    self.pipelines["tts"] = VevoInferencePipeline(
+                        device=self.device,
+                        fmt_cfg_path=self.config_paths["vq8192tomels"],
+                        vocoder_cfg_path=self.config_paths["vocoder"],
+                        ar_cfg_path=self.config_paths["phonetovq8192"],
+                    )
+                except:
+                    # 如果还是失败，创建最简单的管道
+                    self.pipelines["tts"] = VevoInferencePipeline(device=self.device)
             
         return self.pipelines["tts"]