diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index e9ceeb7468..92edefc710 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -587,12 +587,19 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/webis-de/lightning-ir", snippets: snippets.lightning_ir, }, + litert: { + prettyLabel: "LiteRT", + repoName: "LiteRT", + repoUrl: "https://github.com/google-ai-edge/LiteRT", + filter: false, + countDownloads: `path_extension:"tflite"`, + }, "litert-lm": { prettyLabel: "LiteRT-LM", repoName: "LiteRT-LM", repoUrl: "https://github.com/google-ai-edge/LiteRT-LM", filter: false, - countDownloads: `path_extension:"litertlm"`, + countDownloads: `path_extension:"litertlm" OR path_extension:"task"`, }, lerobot: { prettyLabel: "LeRobot",