From 6d4eebe1479d82d317d07dc873c116227a7b2c8b Mon Sep 17 00:00:00 2001 From: Niels Rogge Date: Thu, 19 Jun 2025 14:39:05 +0200 Subject: [PATCH] Add song generation --- packages/tasks/src/model-libraries.ts | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 9a961b8955..c60c1726d1 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -907,6 +907,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { filter: false, countDownloads: `path:"soloaudio_v2.pt"`, }, + "song-generation": { + prettyLabel: "song-generation", + repoName: "song-generation", + repoUrl: "https://github.com/tencent-ailab/songgeneration", + docsUrl: "https://github.com/tencent-ailab/songgeneration", + filter: false, + countDownloads: `path:"ckpt/songgeneration_base/model.pt"`, + }, "stable-baselines3": { prettyLabel: "stable-baselines3", repoName: "stable-baselines3",