Skip to content

Commit de3a986

Browse files
📈 Download tracking for TensorRT model repos (#1783)
When using the `--download-onnx-models` flag with [NVIDIA TensorRT demo/Diffusion tools](https://github.com/NVIDIA/TensorRT/tree/main/demo/Diffusion), ONNX files are downloaded from repos in the https://huggingface.co/collections/stabilityai/nvidia-optimized-684937de2f047a43a8b5456b collection. With this change StabilityAI (and other organizations) will be able to add `library_name: tensorrt` to the model repos to get some statistics on usage. --------- Co-authored-by: Lucain <[email protected]>
1 parent 97ea6a1 commit de3a986

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

packages/tasks/src/model-libraries.ts

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1110,6 +1110,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
11101110
repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
11111111
snippets: snippets.tensorflowtts,
11121112
},
1113+
tensorrt: {
1114+
prettyLabel: "TensorRT",
1115+
repoName: "TensorRT",
1116+
repoUrl: "https://github.com/NVIDIA/TensorRT",
1117+
countDownloads: `path_extension:"onnx"`,
1118+
},
11131119
tabpfn: {
11141120
prettyLabel: "TabPFN",
11151121
repoName: "TabPFN",

0 commit comments

Comments
 (0)