diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 37bb74627..f31dcebb2 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -508,6 +508,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoName: "mindspore", repoUrl: "https://github.com/mindspore-ai/mindspore", }, + "magi-1": { + prettyLabel: "MAGI-1", + repoName: "MAGI-1", + repoUrl: "https://github.com/SandAI-org/MAGI-1", + countDownloads: `path:"ckpt/vae/config.json"`, + }, "mamba-ssm": { prettyLabel: "MambaSSM", repoName: "MambaSSM",