Add TabPFN as library (#1094)

The authors of TabPFN want to track the model (it's being downloaded a lot on GH it seems but not as much on Hub because they track config) @pcuenca --------- Co-authored-by: Pedro Cuenca <[email protected]>
huggingface · Jan 10, 2025 · 74680a1 · 74680a1
1 parent 6a4dddf
commit 74680a1
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
@@ -769,6 +769,12 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 		repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
 		snippets: snippets.tensorflowtts,
 	},
+	tabpfn: {
+		prettyLabel: "TabPFN",
+		repoName: "TabPFN",
+		repoUrl: "https://github.com/PriorLabs/TabPFN",
+		countDownloads: `path_extension:"ckpt" or path:"config.json"`,
+	},
 	"tic-clip": {
 		prettyLabel: "TiC-CLIP",
 		repoName: "TiC-CLIP",