mrfakename commited on
Commit
fa77ee6
·
verified ·
1 Parent(s): fd3ef01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ print("Loading MuCodec decoder...")
52
  mucodec_dir = "./MuCodec"
53
  ckpt_path = os.path.join(mucodec_dir, "ckpt/mucodec.pt")
54
  #audioldm_path = os.path.join(mucodec_dir, "tools/audioldm_48k.pth")
55
- audioldm_path = os.path.join(snapshot_download("haoheliu/audioldm_48k"), "audioldm_48k.pth")
56
  config_path = os.path.join(mucodec_dir, "configs/models/transformer2D.json")
57
 
58
  # Load VAE and STFT
 
52
  mucodec_dir = "./MuCodec"
53
  ckpt_path = os.path.join(mucodec_dir, "ckpt/mucodec.pt")
54
  #audioldm_path = os.path.join(mucodec_dir, "tools/audioldm_48k.pth")
55
+ audioldm_path = os.path.join(snapshot_download("haoheliu/audioldm_48k", local_dir="./alm"), "audioldm_48k.pth")
56
  config_path = os.path.join(mucodec_dir, "configs/models/transformer2D.json")
57
 
58
  # Load VAE and STFT