josephchay commited on
Commit
1b8c0fe
·
1 Parent(s): 70c3139

updated model path

Browse files
Files changed (1) hide show
  1. soundsation/infer/infer_utils.py +1 -1
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)