diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 00638a140c..64c472b80d 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -1096,6 +1096,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS", snippets: snippets.tensorflowtts, }, + tensorrt: { + prettyLabel: "TensorRT", + repoName: "TensorRT", + repoUrl: "https://github.com/NVIDIA/TensorRT", + countDownloads: `path_extension:"onnx"`, + }, tabpfn: { prettyLabel: "TabPFN", repoName: "TabPFN",