Update app.py

app.py

@@ -9,39 +9,26 @@ from huggingface_hub import snapshot_download
 
 # Download All Required Models using `snapshot_download`
 
-def download_and_extract(repo_id, target_dir, cache_dir=None):
-    """
-    Download from HF Hub to cache, then copy to target_dir if not already present.
-    """
-    print(f"Downloading (with cache) {repo_id}...")
-
-    # Use snapshot_download with optional custom cache
-    snapshot_path = snapshot_download(
-        repo_id=repo_id,
-        cache_dir=cache_dir,  # You can pass a shared cache path here
-        local_dir=None,       # Ensure it's using the actual cache
-        local_dir_use_symlinks=False
-    )
-
-    # Copy from cache to target directory
-    if not os.path.exists(target_dir) or not os.listdir(target_dir):
-        os.makedirs(target_dir, exist_ok=True)
-        shutil.copytree(snapshot_path, target_dir, dirs_exist_ok=True)
-        print(f"Copied {repo_id} to {target_dir}")
-    else:
-        print(f"{target_dir} already populated. Skipping copy.")
-
-    return target_dir
-
-
-# Optional: share one cache location across all models
-custom_cache = "./hf_cache"
-
-wan_model_path = download_and_extract("Wan-AI/Wan2.1-I2V-14B-480P", "./weights/Wan2.1-I2V-14B-480P", cache_dir=custom_cache)
-wav2vec_path   = download_and_extract("TencentGameMate/chinese-wav2vec2-base", "./weights/chinese-wav2vec2-base", cache_dir=custom_cache)
-multitalk_path = download_and_extract("MeiGen-AI/MeiGen-MultiTalk", "./weights/MeiGen-MultiTalk", cache_dir=custom_cache)
+# Download Wan2.1-I2V-14B-480P model
+wan_model_path = snapshot_download(
+    repo_id="Wan-AI/Wan2.1-I2V-14B-480P",
+    local_dir="./weights/Wan2.1-I2V-14B-480P",
+    #local_dir_use_symlinks=False
+)
 
+# Download Chinese wav2vec2 model
+wav2vec_path = snapshot_download(
+    repo_id="TencentGameMate/chinese-wav2vec2-base",
+    local_dir="./weights/chinese-wav2vec2-base",
+    #local_dir_use_symlinks=False
+)
 
+# Download MeiGen MultiTalk weights
+multitalk_path = snapshot_download(
+    repo_id="MeiGen-AI/MeiGen-MultiTalk",
+    local_dir="./weights/MeiGen-MultiTalk",
+    #local_dir_use_symlinks=False
+)
 
 # Define paths
 base_model_dir = "./weights/Wan2.1-I2V-14B-480P"
@@ -137,8 +124,8 @@ def infer(prompt, cond_image_path, cond_audio_path):
         "--ckpt_dir", "weights/Wan2.1-I2V-14B-480P",
         "--wav2vec_dir", "weights/chinese-wav2vec2-base",
         "--input_json", "./examples/single_example_1.json",
-        "--sample_steps", "20",
-        "--num_persistent_param_in_dit", str(USED_VRAM_PARAMS),
+        "--sample_steps", "4",
+        "--num_persistent_param_in_dit", str(0),
         "--mode", "streaming",
         "--use_teacache",
         "--save_file", "multi_long_mediumvram_exp"