Skip to content

Commit d24115c

Browse files
tf-marissawWauplin
andauthored
🏷️ Track LiteRT and LiteRT LM files (#1763)
Track LiteRT files (.tflite) and add an additional supported LiteRT LM format for tracking as well. --------- Co-authored-by: Lucain <lucainp@gmail.com>
1 parent 17fb3a7 commit d24115c

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -587,12 +587,19 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
587587
repoUrl: "https://github.com/webis-de/lightning-ir",
588588
snippets: snippets.lightning_ir,
589589
},
590+
litert: {
591+
prettyLabel: "LiteRT",
592+
repoName: "LiteRT",
593+
repoUrl: "https://github.com/google-ai-edge/LiteRT",
594+
filter: false,
595+
countDownloads: `path_extension:"tflite"`,
596+
},
590597
"litert-lm": {
591598
prettyLabel: "LiteRT-LM",
592599
repoName: "LiteRT-LM",
593600
repoUrl: "https://github.com/google-ai-edge/LiteRT-LM",
594601
filter: false,
595-
countDownloads: `path_extension:"litertlm"`,
602+
countDownloads: `path_extension:"litertlm" OR path_extension:"task"`,
596603
},
597604
lerobot: {
598605
prettyLabel: "LeRobot",

0 commit comments

Comments
 (0)