Update app.py

app.py

@@ -46,16 +46,13 @@ def filter_kwargs(cls, kwargs):
 def resolve_path(user_path, repo_root):
     """
     以正確的優先級解析檔案路徑。
-    1. 檢查是否為絕對路徑或本地存在的相對路徑。
-    2. 如果找不到，則嘗試將其視為相對於 Hugging Face 快取根目錄的路徑。
-    3. 如果都不是，則返回 None。
+    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}")
@@ -69,9 +66,10 @@ 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}")
+         raise FileNotFoundError(f"設定檔未找到: {config_path}，請檢查 snapshot_download 是否已下載此檔案。")
     config = OmegaConf.load(config_path)
     sampler_name = "Flow"
     
@@ -203,12 +201,18 @@ def main():
     parser.add_argument('--model_version', type=str, default="square", choices=["square", "rec_vec"], help='StableAvatar 模型版本')
     args = parser.parse_args()
 
-    print("--- 步驟 1: 正在檢查並下載模型 ---")
-    # 這一步仍然是必要的，以確保模型權重檔存在於快取中
+    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/**"],
+        allow_patterns=[
+            "StableAvatar-1.3B/*",
+            "Wan2.1-Fun-V1.1-1.3B-InP/*",
+            "wav2vec2-base-960h/*",
+            "deepspeed_config/**"  # <-- 修正點
+        ],
     )
+    # <<< 修正結束 >>>
     print("模型檔案已準備就緒。")
 
     print("\n--- 步驟 2: 正在解析輸入檔案路徑 ---")
@@ -223,6 +227,7 @@ def main():
         return
 
     print("\n--- 步驟 3: 正在載入模型 ---")
+    # 將 repo_root 傳遞給 setup_models，這樣它才能在正確的位置找到設定檔
     pipeline, transformer3d, vae = setup_models(repo_root, args.model_version)
     print("模型載入完成。")