lshzhm commited on
Commit
18f0bbe
·
1 Parent(s): 923aa33
Files changed (2) hide show
  1. MMAudio/demo.py +1 -1
  2. app.py +13 -0
MMAudio/demo.py CHANGED
@@ -64,7 +64,7 @@ def v2a_load():
64
  if args.variant not in all_model_cfg:
65
  raise ValueError(f'Unknown model variant: {args.variant}')
66
  model: ModelConfig = all_model_cfg[args.variant]
67
- model.download_if_needed()
68
  seq_cfg = model.seq_cfg
69
 
70
  #if args.video:
 
64
  if args.variant not in all_model_cfg:
65
  raise ValueError(f'Unknown model variant: {args.variant}')
66
  model: ModelConfig = all_model_cfg[args.variant]
67
+ #model.download_if_needed()
68
  seq_cfg = model.seq_cfg
69
 
70
  #if args.video:
app.py CHANGED
@@ -31,6 +31,19 @@ if True:
31
  file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="v2c_s44.pt", local_dir=model_path)
32
 
33
  print(f"Model saved at: {file_path}")
 
 
 
 
 
 
 
 
 
 
 
 
 
34
 
35
  log = logging.getLogger()
36
 
 
31
  file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="v2c_s44.pt", local_dir=model_path)
32
 
33
  print(f"Model saved at: {file_path}")
34
+
35
+
36
+ model_path = "./MMAudio/weights/"
37
+
38
+ file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="MMAudio/mmaudio_small_44k.pth", local_dir=model_path)
39
+ print(f"Model saved at: {file_path}")
40
+
41
+ model_path = "./MMAudio/ext_weights/"
42
+
43
+ file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="MMAudio/v1-44.pth", local_dir=model_path)
44
+ print(f"Model saved at: {file_path}")
45
+ file_path = hf_hub_download(repo_id="lshzhm/DeepAudio-V1", filename="MMAudio/synchformer_state_dict.pth", local_dir=model_path)
46
+ print(f"Model saved at: {file_path}")
47
 
48
  log = logging.getLogger()
49