Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
1b8c0fe
1
Parent(s):
70c3139
updated model path
Browse files
soundsation/infer/infer_utils.py
CHANGED
@@ -197,7 +197,7 @@ def encode_audio(audio, vae_model, chunked=False, overlap=32, chunk_size=128):
|
|
197 |
def prepare_model(device):
|
198 |
# prepare cfm model
|
199 |
|
200 |
-
dit_ckpt_path = hf_hub_download(repo_id="josephchay/Soundsation", filename="cfm_model.pt")
|
201 |
dit_config_path = "./soundsation/config/config.json"
|
202 |
with open(dit_config_path) as f:
|
203 |
model_config = json.load(f)
|
|
|
197 |
def prepare_model(device):
|
198 |
# prepare cfm model
|
199 |
|
200 |
+
dit_ckpt_path = hf_hub_download(repo_id="josephchay/Soundsation-giant", filename="cfm_model.pt")
|
201 |
dit_config_path = "./soundsation/config/config.json"
|
202 |
with open(dit_config_path) as f:
|
203 |
model_config = json.load(f)
|