From 3ead236d6ace1dfce338f1080850bfea5d4255a9 Mon Sep 17 00:00:00 2001
From: pharmapsychotic <96542870+pharmapsychotic@users.noreply.github.com>
Date: Fri, 3 Oct 2025 14:04:11 -0500
Subject: [PATCH] Download tracking for HF repos holding ONNX for TensorRT demo
 diffusion models

---
 packages/tasks/src/model-libraries.ts | 6 ++++++
 1 file changed, 6 insertions(+)

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