Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
44845d4
1
Parent(s):
8c1251f
update
Browse files
app.py
CHANGED
|
@@ -188,7 +188,9 @@ def main():
|
|
| 188 |
|
| 189 |
if __name__ == '__main__':
|
| 190 |
model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base-24kHz", "InspireMusic-1.5B-Long-24kHz"]
|
|
|
|
| 191 |
for model_name in model_list:
|
|
|
|
| 192 |
if model_name == "InspireMusic-Base":
|
| 193 |
snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
|
| 194 |
else:
|
|
|
|
| 188 |
|
| 189 |
if __name__ == '__main__':
|
| 190 |
model_list = ["InspireMusic-Base", "InspireMusic-1.5B-Long", "InspireMusic-1.5B", "InspireMusic-Base-24kHz", "InspireMusic-1.5B-Long-24kHz"]
|
| 191 |
+
|
| 192 |
for model_name in model_list:
|
| 193 |
+
model_dir = f"pretrained_models/{model_name}"
|
| 194 |
if model_name == "InspireMusic-Base":
|
| 195 |
snapshot_download(f"iic/InspireMusic", local_dir=model_dir)
|
| 196 |
else:
|