diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index cf218b1131..853faf4e1f 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -1162,6 +1162,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { filter: true, countDownloads: `path:"models/default.zip"`, }, + starflow: { + prettyLabel: "STARFlow", + repoName: "STARFlow", + repoUrl: "https://github.com/apple/ml-starflow", + filter: false, + countDownloads: `path_extension:"pth"`, + }, swarmformer: { prettyLabel: "SwarmFormer", repoName: "SwarmFormer",