diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 92edefc710..664e129379 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -1110,6 +1110,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",