root commited on
Commit
1ca8a57
·
1 Parent(s): c9469d3

downgrade to base model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-large")
20
  os.makedirs(base_full_path, exist_ok=True)
21
- download_model(base_full_path, repo_id="lglg666/SongGeneration-large", revision="75e2043")
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
  # 模型初始化