Skip to content

Commit 6d4eebe

Browse files
committed
Add song generation
1 parent 630483d commit 6d4eebe

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -907,6 +907,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
907907
filter: false,
908908
countDownloads: `path:"soloaudio_v2.pt"`,
909909
},
910+
"song-generation": {
911+
prettyLabel: "song-generation",
912+
repoName: "song-generation",
913+
repoUrl: "https://github.com/tencent-ailab/songgeneration",
914+
docsUrl: "https://github.com/tencent-ailab/songgeneration",
915+
filter: false,
916+
countDownloads: `path:"ckpt/songgeneration_base/model.pt"`,
917+
},
910918
"stable-baselines3": {
911919
prettyLabel: "stable-baselines3",
912920
repoName: "stable-baselines3",

0 commit comments

Comments
 (0)