Add cosmos library (#1089)

This enables downloads [once these models are assigned the `cosmos` library](https://huggingface.co/collections/nvidia/cosmos-6751e884dc10e013a0a0d8e6). Snippets can come later.
huggingface · Jan 8, 2025 · 3c3c008 · 3c3c008
1 parent 4be8b23
commit 3c3c008
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts
@@ -166,6 +166,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		filter: false,
 		countDownloads: `path:"adapter_config.json"`,
 	},
+	cosmos: {
+		prettyLabel: "Cosmos",
+		repoName: "Cosmos",
+		repoUrl: "https://github.com/NVIDIA/Cosmos",
+		countDownloads: `path:"config.json" OR path_extension:"pt"`,
+	},
 	"cxr-foundation": {
 		prettyLabel: "CXR Foundation",
 		repoName: "cxr-foundation",