Spaces:

tencent
/

SongGeneration

Running on L40S

root commited on Oct 31

Commit

4cc99b7

1 Parent(s): f9e2d84

fix bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,12 +16,13 @@ from download import download_model
 # 下载模型
 APP_DIR = op.dirname(op.abspath(__file__))
 download_model(APP_DIR)
-download_model(op.join(APP_DIR, "ckpt"), repo_id="waytan22/SongGeneration-v1.5-beta", revision="db10f47")
 print("Successful downloaded model.")
 # 模型初始化
 from levo_inference import LeVoInference
-MODEL = LeVoInference(op.join(APP_DIR, "ckpt", "SongGeneration-v1.5-beta"))
 EXAMPLE_LYRICS = """
 [intro-medium]

 # 下载模型
 APP_DIR = op.dirname(op.abspath(__file__))
 download_model(APP_DIR)
+large_model_path = op.join(APP_DIR, "ckpt", "SongGeneration-v1.5-beta")
+download_model(large_model_path, repo_id="waytan22/SongGeneration-v1.5-beta", revision="db10f47")
 print("Successful downloaded model.")
 # 模型初始化
 from levo_inference import LeVoInference
+MODEL = LeVoInference(large_model_path)
 EXAMPLE_LYRICS = """
 [intro-medium]