diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index e9ceeb7468..b94acc30ec 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",