Spaces:
Running
on
L40S
Running
on
L40S
root
commited on
Commit
·
1ca8a57
1
Parent(s):
c9469d3
downgrade to base model
Browse files
app.py
CHANGED
|
@@ -16,9 +16,9 @@ from download import download_model
|
|
| 16 |
# 下载模型
|
| 17 |
APP_DIR = op.dirname(op.abspath(__file__))
|
| 18 |
download_model(APP_DIR)
|
| 19 |
-
base_full_path = op.join(APP_DIR, "ckpt", "songgeneration-
|
| 20 |
os.makedirs(base_full_path, exist_ok=True)
|
| 21 |
-
download_model(base_full_path, repo_id="lglg666/SongGeneration-
|
| 22 |
print("Successful downloaded model.")
|
| 23 |
|
| 24 |
# 模型初始化
|
|
|
|
| 16 |
# 下载模型
|
| 17 |
APP_DIR = op.dirname(op.abspath(__file__))
|
| 18 |
download_model(APP_DIR)
|
| 19 |
+
base_full_path = op.join(APP_DIR, "ckpt", "songgeneration-base-full")
|
| 20 |
os.makedirs(base_full_path, exist_ok=True)
|
| 21 |
+
download_model(base_full_path, repo_id="lglg666/SongGeneration-base-full", revision="19ebdb6")
|
| 22 |
print("Successful downloaded model.")
|
| 23 |
|
| 24 |
# 模型初始化
|