doenload model

app.py

@@ -14,28 +14,28 @@ sys.path.append(mmaudio_path)
 from pipeline.pipeline import Pipeline
 from third_party.MMAudio.mmaudio.eval_utils import setup_eval_logging
 
-# # download model
-# os.makedirs("pretrained/mllm", exist_ok=True)
-# from huggingface_hub import snapshot_download
-# repo_local_path = snapshot_download(repo_id="lym0302/VideoLLaMA2.1-7B-AV-CoT", cache_dir='pretrained/mllm')
-
-# remove_vo_model_dir = "pretrained/remove_vo/checkpoints"
-# os.makedirs(remove_vo_model_dir, exist_ok=True)
-# urls = ["https://github.com/TRvlvr/model_repo/releases/download/all_public_uvr_models/model_bs_roformer_ep_317_sdr_12.9755.ckpt",
-#         "https://raw.githubusercontent.com/ZFTurbo/Music-Source-Separation-Training/main/configs/viperx/model_bs_roformer_ep_317_sdr_12.9755.yaml"]
-# for url in urls:
-#     file_name = url.split("/")[-1]  # Extract file name from URL
-#     file_path = os.path.join(remove_vo_model_dir, file_name)
-#     response = requests.get(url, stream=True)
-#     if response.status_code == 200:
-#         with open(file_path, "wb") as f:
-#             for chunk in response.iter_content(chunk_size=8192):  # Use a chunk size of 8 KB
-#                 f.write(chunk)
-#         print(f"File downloaded successfully and saved to {file_path}")
-#     else:
-#         print(f"Failed to download the file. Status code: {response.status_code}")
-
-# os.makedirs("pretrained/v2a/mmaudio", exist_ok=True)
+# download model
+os.makedirs("pretrained/mllm", exist_ok=True)
+from huggingface_hub import snapshot_download
+repo_local_path = snapshot_download(repo_id="lym0302/VideoLLaMA2.1-7B-AV-CoT", cache_dir='pretrained/mllm')
+
+remove_vo_model_dir = "pretrained/remove_vo/checkpoints"
+os.makedirs(remove_vo_model_dir, exist_ok=True)
+urls = ["https://github.com/TRvlvr/model_repo/releases/download/all_public_uvr_models/model_bs_roformer_ep_317_sdr_12.9755.ckpt",
+        "https://raw.githubusercontent.com/ZFTurbo/Music-Source-Separation-Training/main/configs/viperx/model_bs_roformer_ep_317_sdr_12.9755.yaml"]
+for url in urls:
+    file_name = url.split("/")[-1]  # Extract file name from URL
+    file_path = os.path.join(remove_vo_model_dir, file_name)
+    response = requests.get(url, stream=True)
+    if response.status_code == 200:
+        with open(file_path, "wb") as f:
+            for chunk in response.iter_content(chunk_size=8192):  # Use a chunk size of 8 KB
+                f.write(chunk)
+        print(f"File downloaded successfully and saved to {file_path}")
+    else:
+        print(f"Failed to download the file. Status code: {response.status_code}")
+
+os.makedirs("pretrained/v2a/mmaudio", exist_ok=True)
 
 
 setup_eval_logging()