fix

app.py

@@ -82,21 +82,21 @@ def create_args(
     parser.add_argument(
         "--wan_model_dir",
         type=str,
-        default="/data/models/Wan2.1-I2V-14B-720P",
+        default="./models/Wan2.1-I2V-14B-720P",
         required=False,
         help="The dir of the Wan I2V 14B model.",
     )
     parser.add_argument(
         "--fantasytalking_model_path",
         type=str,
-        default="/data/models/fantasytalking_model.ckpt",
+        default="./models/fantasytalking_model.ckpt",
         required=False,
         help="The .ckpt path of fantasytalking model.",
     )
     parser.add_argument(
         "--wav2vec_model_dir",
         type=str,
-        default="/data/models/wav2vec2-base-960h",
+        default="./models/wav2vec2-base-960h",
         required=False,
         help="The dir of wav2vec model.",
     )

infer.py

@@ -125,9 +125,9 @@ def parse_args():
 def load_models(args):
     # Load Wan I2V models
 
-    snapshot_download("Wan-AI/Wan2.1-I2V-14B-720P", local_dir="/data/models/Wan2.1-I2V-14B-720P")
-    snapshot_download("facebook/wav2vec2-base-960h", local_dir="/data/models/wav2vec2-base-960h")
-    snapshot_download("acvlab/FantasyTalking", local_dir="/data/models")
+    snapshot_download("Wan-AI/Wan2.1-I2V-14B-720P", local_dir="./models/Wan2.1-I2V-14B-720P")
+    snapshot_download("facebook/wav2vec2-base-960h", local_dir="./models/wav2vec2-base-960h")
+    snapshot_download("acvlab/FantasyTalking", local_dir="./models")
 
 
     model_manager = ModelManager(device="cpu")