Updated MedVAE Download Counts (#1245)

Hi, thanks for updating the download counts for Merlin. I'm trying to do something similar for MedVAE. The files are under a folder called `model_weights`, but I don't think it matters for `path_extension` since filepaths are inherently flattened. Please let me know if there are any issues! --------- Co-authored-by: Lucain <[email protected]> Co-authored-by: Lucain <[email protected]>
huggingface · Mar 5, 2025 · 4b0963d · 4b0963d
1 parent 1444f66
commit 4b0963d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts
@@ -491,6 +491,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		filter: false,
 		countDownloads: `path_extension:"pt"`,
 	},
+	medvae: {
+		prettyLabel: "MedVAE",
+		repoName: "MedVAE",
+		repoUrl: "https://github.com/StanfordMIMI/MedVAE",
+		filter: false,
+		countDownloads: `path_extension:"ckpt"`,
+	},
 	mitie: {
 		prettyLabel: "MITIE",
 		repoName: "MITIE",