models

MMAudio/demo.py

@@ -64,7 +64,7 @@ def v2a_load():
     if args.variant not in all_model_cfg:
         raise ValueError(f'Unknown model variant: {args.variant}')
     model: ModelConfig = all_model_cfg[args.variant]
-    model.download_if_needed()
+    #model.download_if_needed()
     seq_cfg = model.seq_cfg
 
     #if args.video:

app.py

@@ -31,6 +31,19 @@ if True:
     file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="v2c_s44.pt", local_dir=model_path)
 
     print(f"Model saved at: {file_path}")
+    
+    
+    model_path = "./MMAudio/weights/"
+    
+    file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="MMAudio/mmaudio_small_44k.pth", local_dir=model_path)
+    print(f"Model saved at: {file_path}")
+    
+    model_path = "./MMAudio/ext_weights/"
+    
+    file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="MMAudio/v1-44.pth", local_dir=model_path)
+    print(f"Model saved at: {file_path}")
+    file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="MMAudio/synchformer_state_dict.pth", local_dir=model_path)
+    print(f"Model saved at: {file_path}")
 
 log = logging.getLogger()