hub

app.py

@@ -15,9 +15,9 @@ 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)
+# 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')
+repo_local_path = snapshot_download(repo_id="lym0302/VideoLLaMA2.1-7B-AV-CoT")
 
 remove_vo_model_dir = "pretrained/remove_vo/checkpoints"
 os.makedirs(remove_vo_model_dir, exist_ok=True)
@@ -40,9 +40,9 @@ os.makedirs("pretrained/v2a/mmaudio", exist_ok=True)
 
 setup_eval_logging()
 pipeline = Pipeline(
-    step0_model_dir='pretrained/mllm/models--lym0302--VideoLLaMA2.1-7B-AV-CoT', 
+    step0_model_dir=repo_local_path, 
     step1_mode='mmaudio_medium_44k', 
-    step2_model_dir='pretrained/mllm/models--lym0302--VideoLLaMA2.1-7B-AV-CoT',
+    step2_model_dir=repo_local_path,
     step2_mode='cot',
     step3_mode='bs_roformer',
 )