Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
286fa01
1
Parent(s):
9af664a
update
Browse files
inspiremusic/cli/inference.py
CHANGED
@@ -68,7 +68,8 @@ class InspireMusicUnified:
|
|
68 |
from huggingface_hub import snapshot_download
|
69 |
model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=download_model_dir)
|
70 |
shutil.move(model_dir_tmp, model_dir)
|
71 |
-
|
|
|
72 |
self.model_dir = model_dir
|
73 |
|
74 |
self.sample_rate = sample_rate
|
|
|
68 |
from huggingface_hub import snapshot_download
|
69 |
model_dir_tmp = snapshot_download(repo_id=f"FunAudioLLM/{model_name}", cache_dir=download_model_dir)
|
70 |
shutil.move(model_dir_tmp, model_dir)
|
71 |
+
|
72 |
+
assert os.path.isfile(f"{model_dir}/llm.pt")
|
73 |
self.model_dir = model_dir
|
74 |
|
75 |
self.sample_rate = sample_rate
|