Spaces:
Running
on
Zero
Running
on
Zero
chong.zhang
commited on
Commit
·
c1e0f6f
1
Parent(s):
2f1a479
update
Browse files
inspiremusic/cli/inference.py
CHANGED
@@ -54,7 +54,7 @@ class InspireMusicUnified:
|
|
54 |
|
55 |
# Set model_dir or default to downloading if it doesn't exist
|
56 |
if model_dir is None:
|
57 |
-
model_dir = f"
|
58 |
print(model_dir)
|
59 |
download_model_dir = os.path.dirname(model_dir)
|
60 |
print(download_model_dir)
|
|
|
54 |
|
55 |
# Set model_dir or default to downloading if it doesn't exist
|
56 |
if model_dir is None:
|
57 |
+
model_dir = f"../../pretrained_models/{model_name}"
|
58 |
print(model_dir)
|
59 |
download_model_dir = os.path.dirname(model_dir)
|
60 |
print(download_model_dir)
|