Download tracking for HF repos holding ONNX for TensorRT demo diffusion models

pharmapsychotic · pharmapsychotic · commit 3ead236d6ace · 2025-10-03T14:04:11.000-05:00
diff --git 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",
