@huggingface/tasks 0.13.1 → 0.13.2

package/dist/commonjs/model-libraries.js

@@ -0,0 +1,793 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ALL_DISPLAY_MODEL_LIBRARY_KEYS = exports.ALL_MODEL_LIBRARY_KEYS = exports.MODEL_LIBRARIES_UI_ELEMENTS = void 0;
+const snippets = __importStar(require("./model-libraries-snippets.js"));
+/**
+ * Add your new library here.
+ *
+ * This is for modeling (= architectures) libraries, not for file formats (like ONNX, etc).
+ * (unlike libraries, file formats live in an enum inside the internal codebase.)
+ *
+ * Doc on how to add a library to the Hub:
+ *
+ * https://huggingface.co/docs/hub/models-adding-libraries
+ *
+ * /!\ IMPORTANT
+ *
+ * The key you choose is the tag your models have in their library_name on the Hub.
+ */
+exports.MODEL_LIBRARIES_UI_ELEMENTS = {
+    "adapter-transformers": {
+        prettyLabel: "Adapters",
+        repoName: "adapters",
+        repoUrl: "https://github.com/Adapter-Hub/adapters",
+        docsUrl: "https://huggingface.co/docs/hub/adapters",
+        snippets: snippets.adapters,
+        filter: true,
+        countDownloads: `path:"adapter_config.json"`,
+    },
+    allennlp: {
+        prettyLabel: "AllenNLP",
+        repoName: "AllenNLP",
+        repoUrl: "https://github.com/allenai/allennlp",
+        docsUrl: "https://huggingface.co/docs/hub/allennlp",
+        snippets: snippets.allennlp,
+        filter: true,
+    },
+    asteroid: {
+        prettyLabel: "Asteroid",
+        repoName: "Asteroid",
+        repoUrl: "https://github.com/asteroid-team/asteroid",
+        docsUrl: "https://huggingface.co/docs/hub/asteroid",
+        snippets: snippets.asteroid,
+        filter: true,
+        countDownloads: `path:"pytorch_model.bin"`,
+    },
+    audiocraft: {
+        prettyLabel: "Audiocraft",
+        repoName: "audiocraft",
+        repoUrl: "https://github.com/facebookresearch/audiocraft",
+        snippets: snippets.audiocraft,
+        filter: false,
+        countDownloads: `path:"state_dict.bin"`,
+    },
+    audioseal: {
+        prettyLabel: "AudioSeal",
+        repoName: "audioseal",
+        repoUrl: "https://github.com/facebookresearch/audioseal",
+        filter: false,
+        countDownloads: `path_extension:"pth"`,
+        snippets: snippets.audioseal,
+    },
+    bertopic: {
+        prettyLabel: "BERTopic",
+        repoName: "BERTopic",
+        repoUrl: "https://github.com/MaartenGr/BERTopic",
+        snippets: snippets.bertopic,
+        filter: true,
+    },
+    big_vision: {
+        prettyLabel: "Big Vision",
+        repoName: "big_vision",
+        repoUrl: "https://github.com/google-research/big_vision",
+        filter: false,
+        countDownloads: `path_extension:"npz"`,
+    },
+    birefnet: {
+        prettyLabel: "BiRefNet",
+        repoName: "BiRefNet",
+        repoUrl: "https://github.com/ZhengPeng7/BiRefNet",
+        snippets: snippets.birefnet,
+        filter: false,
+    },
+    bm25s: {
+        prettyLabel: "BM25S",
+        repoName: "bm25s",
+        repoUrl: "https://github.com/xhluca/bm25s",
+        snippets: snippets.bm25s,
+        filter: false,
+        countDownloads: `path:"params.index.json"`,
+    },
+    champ: {
+        prettyLabel: "Champ",
+        repoName: "Champ",
+        repoUrl: "https://github.com/fudan-generative-vision/champ",
+        countDownloads: `path:"champ/motion_module.pth"`,
+    },
+    chat_tts: {
+        prettyLabel: "ChatTTS",
+        repoName: "ChatTTS",
+        repoUrl: "https://github.com/2noise/ChatTTS.git",
+        snippets: snippets.chattts,
+        filter: false,
+        countDownloads: `path:"asset/GPT.pt"`,
+    },
+    colpali: {
+        prettyLabel: "ColPali",
+        repoName: "ColPali",
+        repoUrl: "https://github.com/ManuelFay/colpali",
+        filter: false,
+        countDownloads: `path:"adapter_config.json"`,
+    },
+    deepforest: {
+        prettyLabel: "DeepForest",
+        repoName: "deepforest",
+        docsUrl: "https://deepforest.readthedocs.io/en/latest/",
+        repoUrl: "https://github.com/weecology/DeepForest",
+        countDownloads: `path_extension:"pt" OR path_extension:"pl"`,
+    },
+    "depth-anything-v2": {
+        prettyLabel: "DepthAnythingV2",
+        repoName: "Depth Anything V2",
+        repoUrl: "https://github.com/DepthAnything/Depth-Anything-V2",
+        snippets: snippets.depth_anything_v2,
+        filter: false,
+        countDownloads: `path_extension:"pth"`,
+    },
+    "depth-pro": {
+        prettyLabel: "Depth Pro",
+        repoName: "Depth Pro",
+        repoUrl: "https://github.com/apple/ml-depth-pro",
+        countDownloads: `path_extension:"pt"`,
+        snippets: snippets.depth_pro,
+        filter: false,
+    },
+    diffree: {
+        prettyLabel: "Diffree",
+        repoName: "Diffree",
+        repoUrl: "https://github.com/OpenGVLab/Diffree",
+        filter: false,
+        countDownloads: `path:"diffree-step=000010999.ckpt"`,
+    },
+    diffusers: {
+        prettyLabel: "Diffusers",
+        repoName: "🤗/diffusers",
+        repoUrl: "https://github.com/huggingface/diffusers",
+        docsUrl: "https://huggingface.co/docs/hub/diffusers",
+        snippets: snippets.diffusers,
+        filter: true,
+        /// diffusers has its own more complex "countDownloads" query
+    },
+    diffusionkit: {
+        prettyLabel: "DiffusionKit",
+        repoName: "DiffusionKit",
+        repoUrl: "https://github.com/argmaxinc/DiffusionKit",
+        snippets: snippets.diffusionkit,
+    },
+    doctr: {
+        prettyLabel: "docTR",
+        repoName: "doctr",
+        repoUrl: "https://github.com/mindee/doctr",
+    },
+    cartesia_pytorch: {
+        prettyLabel: "Cartesia Pytorch",
+        repoName: "Cartesia Pytorch",
+        repoUrl: "https://github.com/cartesia-ai/cartesia_pytorch",
+        snippets: snippets.cartesia_pytorch,
+    },
+    cartesia_mlx: {
+        prettyLabel: "Cartesia MLX",
+        repoName: "Cartesia MLX",
+        repoUrl: "https://github.com/cartesia-ai/cartesia_mlx",
+        snippets: snippets.cartesia_mlx,
+    },
+    cotracker: {
+        prettyLabel: "CoTracker",
+        repoName: "CoTracker",
+        repoUrl: "https://github.com/facebookresearch/co-tracker",
+        filter: false,
+        countDownloads: `path_extension:"pth"`,
+    },
+    edsnlp: {
+        prettyLabel: "EDS-NLP",
+        repoName: "edsnlp",
+        repoUrl: "https://github.com/aphp/edsnlp",
+        docsUrl: "https://aphp.github.io/edsnlp/latest/",
+        filter: false,
+        snippets: snippets.edsnlp,
+        countDownloads: `path_filename:"config" AND path_extension:"cfg"`,
+    },
+    elm: {
+        prettyLabel: "ELM",
+        repoName: "elm",
+        repoUrl: "https://github.com/slicex-ai/elm",
+        filter: false,
+        countDownloads: `path_filename:"slicex_elm_config" AND path_extension:"json"`,
+    },
+    espnet: {
+        prettyLabel: "ESPnet",
+        repoName: "ESPnet",
+        repoUrl: "https://github.com/espnet/espnet",
+        docsUrl: "https://huggingface.co/docs/hub/espnet",
+        snippets: snippets.espnet,
+        filter: true,
+    },
+    fairseq: {
+        prettyLabel: "Fairseq",
+        repoName: "fairseq",
+        repoUrl: "https://github.com/pytorch/fairseq",
+        snippets: snippets.fairseq,
+        filter: true,
+    },
+    fastai: {
+        prettyLabel: "fastai",
+        repoName: "fastai",
+        repoUrl: "https://github.com/fastai/fastai",
+        docsUrl: "https://huggingface.co/docs/hub/fastai",
+        snippets: snippets.fastai,
+        filter: true,
+    },
+    fasttext: {
+        prettyLabel: "fastText",
+        repoName: "fastText",
+        repoUrl: "https://fasttext.cc/",
+        snippets: snippets.fasttext,
+        filter: true,
+        countDownloads: `path_extension:"bin"`,
+    },
+    flair: {
+        prettyLabel: "Flair",
+        repoName: "Flair",
+        repoUrl: "https://github.com/flairNLP/flair",
+        docsUrl: "https://huggingface.co/docs/hub/flair",
+        snippets: snippets.flair,
+        filter: true,
+        countDownloads: `path:"pytorch_model.bin"`,
+    },
+    "gemma.cpp": {
+        prettyLabel: "gemma.cpp",
+        repoName: "gemma.cpp",
+        repoUrl: "https://github.com/google/gemma.cpp",
+        filter: false,
+        countDownloads: `path_extension:"sbs"`,
+    },
+    gliner: {
+        prettyLabel: "GLiNER",
+        repoName: "GLiNER",
+        repoUrl: "https://github.com/urchade/GLiNER",
+        snippets: snippets.gliner,
+        filter: false,
+        countDownloads: `path:"gliner_config.json"`,
+    },
+    "glyph-byt5": {
+        prettyLabel: "Glyph-ByT5",
+        repoName: "Glyph-ByT5",
+        repoUrl: "https://github.com/AIGText/Glyph-ByT5",
+        filter: false,
+        countDownloads: `path:"checkpoints/byt5_model.pt"`,
+    },
+    grok: {
+        prettyLabel: "Grok",
+        repoName: "Grok",
+        repoUrl: "https://github.com/xai-org/grok-1",
+        filter: false,
+        countDownloads: `path:"ckpt/tensor00000_000" OR path:"ckpt-0/tensor00000_000"`,
+    },
+    hallo: {
+        prettyLabel: "Hallo",
+        repoName: "Hallo",
+        repoUrl: "https://github.com/fudan-generative-vision/hallo",
+        countDownloads: `path:"hallo/net.pth"`,
+    },
+    hezar: {
+        prettyLabel: "Hezar",
+        repoName: "Hezar",
+        repoUrl: "https://github.com/hezarai/hezar",
+        docsUrl: "https://hezarai.github.io/hezar",
+        countDownloads: `path:"model_config.yaml" OR path:"embedding/embedding_config.yaml"`,
+    },
+    htrflow: {
+        prettyLabel: "HTRflow",
+        repoName: "HTRflow",
+        repoUrl: "https://github.com/AI-Riksarkivet/htrflow",
+        docsUrl: "https://ai-riksarkivet.github.io/htrflow",
+        snippets: snippets.htrflow,
+    },
+    "hunyuan-dit": {
+        prettyLabel: "HunyuanDiT",
+        repoName: "HunyuanDiT",
+        repoUrl: "https://github.com/Tencent/HunyuanDiT",
+        countDownloads: `path:"pytorch_model_ema.pt" OR path:"pytorch_model_distill.pt"`,
+    },
+    imstoucan: {
+        prettyLabel: "IMS Toucan",
+        repoName: "IMS-Toucan",
+        repoUrl: "https://github.com/DigitalPhonetics/IMS-Toucan",
+        countDownloads: `path:"embedding_gan.pt" OR path:"Vocoder.pt" OR path:"ToucanTTS.pt"`,
+    },
+    keras: {
+        prettyLabel: "Keras",
+        repoName: "Keras",
+        repoUrl: "https://github.com/keras-team/keras",
+        docsUrl: "https://huggingface.co/docs/hub/keras",
+        snippets: snippets.keras,
+        filter: true,
+        countDownloads: `path:"config.json" OR path_extension:"keras"`,
+    },
+    "tf-keras": {
+        // Legacy "Keras 2" library (tensorflow-only)
+        prettyLabel: "TF-Keras",
+        repoName: "TF-Keras",
+        repoUrl: "https://github.com/keras-team/tf-keras",
+        docsUrl: "https://huggingface.co/docs/hub/tf-keras",
+        snippets: snippets.tf_keras,
+        countDownloads: `path:"saved_model.pb"`,
+    },
+    "keras-nlp": {
+        prettyLabel: "KerasNLP",
+        repoName: "KerasNLP",
+        repoUrl: "https://github.com/keras-team/keras-nlp",
+        docsUrl: "https://keras.io/keras_nlp/",
+        snippets: snippets.keras_nlp,
+    },
+    "keras-hub": {
+        prettyLabel: "KerasHub",
+        repoName: "KerasHub",
+        repoUrl: "https://github.com/keras-team/keras-hub",
+        docsUrl: "https://keras.io/keras_hub/",
+        snippets: snippets.keras_hub,
+        filter: true,
+    },
+    k2: {
+        prettyLabel: "K2",
+        repoName: "k2",
+        repoUrl: "https://github.com/k2-fsa/k2",
+    },
+    liveportrait: {
+        prettyLabel: "LivePortrait",
+        repoName: "LivePortrait",
+        repoUrl: "https://github.com/KwaiVGI/LivePortrait",
+        filter: false,
+        countDownloads: `path:"liveportrait/landmark.onnx"`,
+    },
+    "llama-cpp-python": {
+        prettyLabel: "llama-cpp-python",
+        repoName: "llama-cpp-python",
+        repoUrl: "https://github.com/abetlen/llama-cpp-python",
+        snippets: snippets.llama_cpp_python,
+    },
+    "mini-omni2": {
+        prettyLabel: "Mini-Omni2",
+        repoName: "Mini-Omni2",
+        repoUrl: "https://github.com/gpt-omni/mini-omni2",
+        countDownloads: `path:"model_config.yaml"`,
+    },
+    mindspore: {
+        prettyLabel: "MindSpore",
+        repoName: "mindspore",
+        repoUrl: "https://github.com/mindspore-ai/mindspore",
+    },
+    "mamba-ssm": {
+        prettyLabel: "MambaSSM",
+        repoName: "MambaSSM",
+        repoUrl: "https://github.com/state-spaces/mamba",
+        filter: false,
+        snippets: snippets.mamba_ssm,
+    },
+    "mars5-tts": {
+        prettyLabel: "MARS5-TTS",
+        repoName: "MARS5-TTS",
+        repoUrl: "https://github.com/Camb-ai/MARS5-TTS",
+        filter: false,
+        countDownloads: `path:"mars5_ar.safetensors"`,
+        snippets: snippets.mars5_tts,
+    },
+    "mesh-anything": {
+        prettyLabel: "MeshAnything",
+        repoName: "MeshAnything",
+        repoUrl: "https://github.com/buaacyw/MeshAnything",
+        filter: false,
+        countDownloads: `path:"MeshAnything_350m.pth"`,
+        snippets: snippets.mesh_anything,
+    },
+    "ml-agents": {
+        prettyLabel: "ml-agents",
+        repoName: "ml-agents",
+        repoUrl: "https://github.com/Unity-Technologies/ml-agents",
+        docsUrl: "https://huggingface.co/docs/hub/ml-agents",
+        snippets: snippets.mlAgents,
+        filter: true,
+        countDownloads: `path_extension:"onnx"`,
+    },
+    mlx: {
+        prettyLabel: "MLX",
+        repoName: "MLX",
+        repoUrl: "https://github.com/ml-explore/mlx-examples/tree/main",
+        snippets: snippets.mlx,
+        filter: true,
+    },
+    "mlx-image": {
+        prettyLabel: "mlx-image",
+        repoName: "mlx-image",
+        repoUrl: "https://github.com/riccardomusmeci/mlx-image",
+        docsUrl: "https://huggingface.co/docs/hub/mlx-image",
+        snippets: snippets.mlxim,
+        filter: false,
+        countDownloads: `path:"model.safetensors"`,
+    },
+    "mlc-llm": {
+        prettyLabel: "MLC-LLM",
+        repoName: "MLC-LLM",
+        repoUrl: "https://github.com/mlc-ai/mlc-llm",
+        docsUrl: "https://llm.mlc.ai/docs/",
+        filter: false,
+        countDownloads: `path:"mlc-chat-config.json"`,
+    },
+    model2vec: {
+        prettyLabel: "Model2Vec",
+        repoName: "model2vec",
+        repoUrl: "https://github.com/MinishLab/model2vec",
+        snippets: snippets.model2vec,
+        filter: false,
+    },
+    moshi: {
+        prettyLabel: "Moshi",
+        repoName: "Moshi",
+        repoUrl: "https://github.com/kyutai-labs/moshi",
+        filter: false,
+        countDownloads: `path:"tokenizer-e351c8d8-checkpoint125.safetensors"`,
+    },
+    nemo: {
+        prettyLabel: "NeMo",
+        repoName: "NeMo",
+        repoUrl: "https://github.com/NVIDIA/NeMo",
+        snippets: snippets.nemo,
+        filter: true,
+        countDownloads: `path_extension:"nemo" OR path:"model_config.yaml"`,
+    },
+    "open-oasis": {
+        prettyLabel: "open-oasis",
+        repoName: "open-oasis",
+        repoUrl: "https://github.com/etched-ai/open-oasis",
+        countDownloads: `path:"oasis500m.safetensors"`,
+    },
+    open_clip: {
+        prettyLabel: "OpenCLIP",
+        repoName: "OpenCLIP",
+        repoUrl: "https://github.com/mlfoundations/open_clip",
+        snippets: snippets.open_clip,
+        filter: true,
+        countDownloads: `path_extension:"bin" AND path_filename:*pytorch_model`,
+    },
+    paddlenlp: {
+        prettyLabel: "paddlenlp",
+        repoName: "PaddleNLP",
+        repoUrl: "https://github.com/PaddlePaddle/PaddleNLP",
+        docsUrl: "https://huggingface.co/docs/hub/paddlenlp",
+        snippets: snippets.paddlenlp,
+        filter: true,
+        countDownloads: `path:"model_config.json"`,
+    },
+    peft: {
+        prettyLabel: "PEFT",
+        repoName: "PEFT",
+        repoUrl: "https://github.com/huggingface/peft",
+        snippets: snippets.peft,
+        filter: true,
+        countDownloads: `path:"adapter_config.json"`,
+    },
+    pxia: {
+        prettyLabel: "pxia",
+        repoName: "pxia",
+        repoUrl: "https://github.com/not-lain/pxia",
+        snippets: snippets.pxia,
+        filter: false,
+    },
+    "pyannote-audio": {
+        prettyLabel: "pyannote.audio",
+        repoName: "pyannote-audio",
+        repoUrl: "https://github.com/pyannote/pyannote-audio",
+        snippets: snippets.pyannote_audio,
+        filter: true,
+    },
+    "py-feat": {
+        prettyLabel: "Py-Feat",
+        repoName: "Py-Feat",
+        repoUrl: "https://github.com/cosanlab/py-feat",
+        docsUrl: "https://py-feat.org/",
+        filter: false,
+    },
+    pythae: {
+        prettyLabel: "pythae",
+        repoName: "pythae",
+        repoUrl: "https://github.com/clementchadebec/benchmark_VAE",
+        snippets: snippets.pythae,
+        filter: false,
+    },
+    recurrentgemma: {
+        prettyLabel: "RecurrentGemma",
+        repoName: "recurrentgemma",
+        repoUrl: "https://github.com/google-deepmind/recurrentgemma",
+        filter: false,
+        countDownloads: `path:"tokenizer.model"`,
+    },
+    relik: {
+        prettyLabel: "Relik",
+        repoName: "Relik",
+        repoUrl: "https://github.com/SapienzaNLP/relik",
+        snippets: snippets.relik,
+        filter: false,
+    },
+    refiners: {
+        prettyLabel: "Refiners",
+        repoName: "Refiners",
+        repoUrl: "https://github.com/finegrain-ai/refiners",
+        docsUrl: "https://refine.rs/",
+        filter: false,
+        countDownloads: `path:"model.safetensors"`,
+    },
+    reverb: {
+        prettyLabel: "Reverb",
+        repoName: "Reverb",
+        repoUrl: "https://github.com/revdotcom/reverb",
+        filter: false,
+    },
+    saelens: {
+        prettyLabel: "SAELens",
+        repoName: "SAELens",
+        repoUrl: "https://github.com/jbloomAus/SAELens",
+        snippets: snippets.saelens,
+        filter: false,
+    },
+    sam2: {
+        prettyLabel: "sam2",
+        repoName: "sam2",
+        repoUrl: "https://github.com/facebookresearch/segment-anything-2",
+        filter: false,
+        snippets: snippets.sam2,
+        countDownloads: `path_extension:"pt"`,
+    },
+    "sample-factory": {
+        prettyLabel: "sample-factory",
+        repoName: "sample-factory",
+        repoUrl: "https://github.com/alex-petrenko/sample-factory",
+        docsUrl: "https://huggingface.co/docs/hub/sample-factory",
+        snippets: snippets.sampleFactory,
+        filter: true,
+        countDownloads: `path:"cfg.json"`,
+    },
+    sapiens: {
+        prettyLabel: "sapiens",
+        repoName: "sapiens",
+        repoUrl: "https://github.com/facebookresearch/sapiens",
+        filter: false,
+        countDownloads: `path_extension:"pt2" OR path_extension:"pth" OR path_extension:"onnx"`,
+    },
+    "sentence-transformers": {
+        prettyLabel: "sentence-transformers",
+        repoName: "sentence-transformers",
+        repoUrl: "https://github.com/UKPLab/sentence-transformers",
+        docsUrl: "https://huggingface.co/docs/hub/sentence-transformers",
+        snippets: snippets.sentenceTransformers,
+        filter: true,
+    },
+    setfit: {
+        prettyLabel: "setfit",
+        repoName: "setfit",
+        repoUrl: "https://github.com/huggingface/setfit",
+        docsUrl: "https://huggingface.co/docs/hub/setfit",
+        snippets: snippets.setfit,
+        filter: true,
+    },
+    sklearn: {
+        prettyLabel: "Scikit-learn",
+        repoName: "Scikit-learn",
+        repoUrl: "https://github.com/scikit-learn/scikit-learn",
+        snippets: snippets.sklearn,
+        filter: true,
+        countDownloads: `path:"sklearn_model.joblib"`,
+    },
+    spacy: {
+        prettyLabel: "spaCy",
+        repoName: "spaCy",
+        repoUrl: "https://github.com/explosion/spaCy",
+        docsUrl: "https://huggingface.co/docs/hub/spacy",
+        snippets: snippets.spacy,
+        filter: true,
+        countDownloads: `path_extension:"whl"`,
+    },
+    "span-marker": {
+        prettyLabel: "SpanMarker",
+        repoName: "SpanMarkerNER",
+        repoUrl: "https://github.com/tomaarsen/SpanMarkerNER",
+        docsUrl: "https://huggingface.co/docs/hub/span_marker",
+        snippets: snippets.span_marker,
+        filter: true,
+    },
+    speechbrain: {
+        prettyLabel: "speechbrain",
+        repoName: "speechbrain",
+        repoUrl: "https://github.com/speechbrain/speechbrain",
+        docsUrl: "https://huggingface.co/docs/hub/speechbrain",
+        snippets: snippets.speechbrain,
+        filter: true,
+        countDownloads: `path:"hyperparams.yaml"`,
+    },
+    "ssr-speech": {
+        prettyLabel: "SSR-Speech",
+        repoName: "SSR-Speech",
+        repoUrl: "https://github.com/WangHelin1997/SSR-Speech",
+        filter: false,
+        countDownloads: `path_extension:".pth"`,
+    },
+    "stable-audio-tools": {
+        prettyLabel: "Stable Audio Tools",
+        repoName: "stable-audio-tools",
+        repoUrl: "https://github.com/Stability-AI/stable-audio-tools.git",
+        filter: false,
+        countDownloads: `path:"model.safetensors"`,
+        snippets: snippets.stable_audio_tools,
+    },
+    "diffusion-single-file": {
+        prettyLabel: "Diffusion Single File",
+        repoName: "diffusion-single-file",
+        repoUrl: "https://github.com/comfyanonymous/ComfyUI",
+        filter: false,
+        countDownloads: `path_extension:"safetensors"`,
+    },
+    "seed-story": {
+        prettyLabel: "SEED-Story",
+        repoName: "SEED-Story",
+        repoUrl: "https://github.com/TencentARC/SEED-Story",
+        filter: false,
+        countDownloads: `path:"cvlm_llama2_tokenizer/tokenizer.model"`,
+        snippets: snippets.seed_story,
+    },
+    soloaudio: {
+        prettyLabel: "SoloAudio",
+        repoName: "SoloAudio",
+        repoUrl: "https://github.com/WangHelin1997/SoloAudio",
+        filter: false,
+        countDownloads: `path:"soloaudio_v2.pt"`,
+    },
+    "stable-baselines3": {
+        prettyLabel: "stable-baselines3",
+        repoName: "stable-baselines3",
+        repoUrl: "https://github.com/huggingface/huggingface_sb3",
+        docsUrl: "https://huggingface.co/docs/hub/stable-baselines3",
+        snippets: snippets.stableBaselines3,
+        filter: true,
+        countDownloads: `path_extension:"zip"`,
+    },
+    stanza: {
+        prettyLabel: "Stanza",
+        repoName: "stanza",
+        repoUrl: "https://github.com/stanfordnlp/stanza",
+        docsUrl: "https://huggingface.co/docs/hub/stanza",
+        snippets: snippets.stanza,
+        filter: true,
+        countDownloads: `path:"models/default.zip"`,
+    },
+    "f5-tts": {
+        prettyLabel: "F5-TTS",
+        repoName: "F5-TTS",
+        repoUrl: "https://github.com/SWivid/F5-TTS",
+        filter: false,
+        countDownloads: `path_extension:"safetensors" OR path_extension:"pt"`,
+    },
+    genmo: {
+        prettyLabel: "Genmo",
+        repoName: "Genmo",
+        repoUrl: "https://github.com/genmoai/models",
+        filter: false,
+        countDownloads: `path:"vae_stats.json"`,
+    },
+    tensorflowtts: {
+        prettyLabel: "TensorFlowTTS",
+        repoName: "TensorFlowTTS",
+        repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
+        snippets: snippets.tensorflowtts,
+    },
+    "tic-clip": {
+        prettyLabel: "TiC-CLIP",
+        repoName: "TiC-CLIP",
+        repoUrl: "https://github.com/apple/ml-tic-clip",
+        filter: false,
+        countDownloads: `path_extension:"pt" AND path_prefix:"checkpoints/"`,
+    },
+    timesfm: {
+        prettyLabel: "TimesFM",
+        repoName: "timesfm",
+        repoUrl: "https://github.com/google-research/timesfm",
+        filter: false,
+        countDownloads: `path:"checkpoints/checkpoint_1100000/state/checkpoint"`,
+    },
+    timm: {
+        prettyLabel: "timm",
+        repoName: "pytorch-image-models",
+        repoUrl: "https://github.com/rwightman/pytorch-image-models",
+        docsUrl: "https://huggingface.co/docs/hub/timm",
+        snippets: snippets.timm,
+        filter: true,
+        countDownloads: `path:"pytorch_model.bin" OR path:"model.safetensors"`,
+    },
+    transformers: {
+        prettyLabel: "Transformers",
+        repoName: "🤗/transformers",
+        repoUrl: "https://github.com/huggingface/transformers",
+        docsUrl: "https://huggingface.co/docs/hub/transformers",
+        snippets: snippets.transformers,
+        filter: true,
+    },
+    "transformers.js": {
+        prettyLabel: "Transformers.js",
+        repoName: "transformers.js",
+        repoUrl: "https://github.com/huggingface/transformers.js",
+        docsUrl: "https://huggingface.co/docs/hub/transformers-js",
+        snippets: snippets.transformersJS,
+        filter: true,
+    },
+    "unity-sentis": {
+        prettyLabel: "unity-sentis",
+        repoName: "unity-sentis",
+        repoUrl: "https://github.com/Unity-Technologies/sentis-samples",
+        snippets: snippets.sentis,
+        filter: true,
+        countDownloads: `path_extension:"sentis"`,
+    },
+    "vfi-mamba": {
+        prettyLabel: "VFIMamba",
+        repoName: "VFIMamba",
+        repoUrl: "https://github.com/MCG-NJU/VFIMamba",
+        countDownloads: `path_extension:"pkl"`,
+        snippets: snippets.vfimamba,
+    },
+    voicecraft: {
+        prettyLabel: "VoiceCraft",
+        repoName: "VoiceCraft",
+        repoUrl: "https://github.com/jasonppy/VoiceCraft",
+        docsUrl: "https://github.com/jasonppy/VoiceCraft",
+        snippets: snippets.voicecraft,
+    },
+    yolov10: {
+        prettyLabel: "YOLOv10",
+        repoName: "yolov10",
+        repoUrl: "https://github.com/THU-MIG/yolov10",
+        docsUrl: "https://github.com/THU-MIG/yolov10",
+        snippets: snippets.yolov10,
+    },
+    whisperkit: {
+        prettyLabel: "WhisperKit",
+        repoName: "WhisperKit",
+        repoUrl: "https://github.com/argmaxinc/WhisperKit",
+        docsUrl: "https://github.com/argmaxinc/WhisperKit?tab=readme-ov-file#homebrew",
+        snippets: snippets.whisperkit,
+        countDownloads: `path_filename:"model" AND path_extension:"mil" AND _exists_:"path_prefix"`,
+    },
+    "3dtopia-xl": {
+        prettyLabel: "3DTopia-XL",
+        repoName: "3DTopia-XL",
+        repoUrl: "https://github.com/3DTopia/3DTopia-XL",
+        filter: false,
+        countDownloads: `path:"model_vae_fp16.pt"`,
+        snippets: snippets.threedtopia_xl,
+    },
+};
+exports.ALL_MODEL_LIBRARY_KEYS = Object.keys(exports.MODEL_LIBRARIES_UI_ELEMENTS);
+exports.ALL_DISPLAY_MODEL_LIBRARY_KEYS = Object.entries(exports.MODEL_LIBRARIES_UI_ELEMENTS)
+    // eslint-disable-next-line @typescript-eslint/no-unused-vars
+    .filter(([_, v]) => v.filter)
+    .map(([k]) => k);