Update app.py
Browse files
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
|