Update app.py

app.py

@@ -49,16 +49,17 @@ def resolve_path(user_path, repo_root):
     1. 優先檢查本地路徑（絕對或相對）。
     2. 如果找不到，則嘗試從 HF 快取目錄中尋找。
     """
+    # 優先檢查本地路徑
     if os.path.exists(user_path):
         print(f"找到本地檔案： {os.path.abspath(user_path)}")
         return os.path.abspath(user_path)
     
+    # 其次，嘗試從 HF 快取目錄中尋找
     potential_repo_path = os.path.join(repo_root, user_path)
     if os.path.exists(potential_repo_path):
         print(f"在 Hugging Face 快取目錄中找到檔案： {potential_repo_path}")
         return potential_repo_path
         
-    print(f"錯誤：在任何位置都找不到檔案： {user_path}")
     return None
 
 def setup_models(repo_root, model_version):
@@ -66,10 +67,15 @@ def setup_models(repo_root, model_version):
     pretrained_model_name_or_path = os.path.join(repo_root, "Wan2.1-Fun-V1.1-1.3B-InP")
     pretrained_wav2vec_path = os.path.join(repo_root, "wav2vec2-base-960h")
 
-    # 這個路徑現在應該可以正確找到了
-    config_path = os.path.join(repo_root, "deepspeed_config/wan2.1/wan_civitai.yaml")
-    if not os.path.exists(config_path):
-         raise FileNotFoundError(f"設定檔未找到: {config_path}，請檢查 snapshot_download 是否已下載此檔案。")
+    # <<< 核心修正：對設定檔使用與輸入檔案相同的路徑解析邏輯 >>>
+    config_relative_path = "deepspeed_config/wan2.1/wan_civitai.yaml"
+    config_path = resolve_path(config_relative_path, repo_root)
+    
+    if not config_path:
+        raise FileNotFoundError(f"設定檔 '{config_relative_path}' 在當前目錄或 HF 快取中都找不到。請確保該檔案存在。")
+    # <<< 修正結束 >>>
+
+    print(f"正在從 {config_path} 載入設定...")
     config = OmegaConf.load(config_path)
     sampler_name = "Flow"
     
@@ -202,17 +208,17 @@ def main():
     args = parser.parse_args()
 
     print("--- 步驟 1: 正在檢查並下載模型與設定檔 ---")
-    # <<< 核心修正：加入 'deepspeed_config/**' 來下載設定檔 >>>
+    # 確保所有需要的檔案都被下載，以作為本地找不到檔案時的後備
     repo_root = snapshot_download(
         repo_id="FrancisRing/StableAvatar",
         allow_patterns=[
             "StableAvatar-1.3B/*",
             "Wan2.1-Fun-V1.1-1.3B-InP/*",
             "wav2vec2-base-960h/*",
-            "deepspeed_config/**"  # <-- 修正點
+            "deepspeed_config/**",
+            "example_case/**" # 也下載範例，以防使用者直接執行預設參數
         ],
     )
-    # <<< 修正結束 >>>
     print("模型檔案已準備就緒。")
 
     print("\n--- 步驟 2: 正在解析輸入檔案路徑 ---")
@@ -227,7 +233,6 @@ def main():
         return
 
     print("\n--- 步驟 3: 正在載入模型 ---")
-    # 將 repo_root 傳遞給 setup_models，這樣它才能在正確的位置找到設定檔
     pipeline, transformer3d, vae = setup_models(repo_root, args.model_version)
     print("模型載入完成。")