@huggingface/tasks
Version:
List of ML tasks for huggingface.co/tasks
1,270 lines (1,269 loc) • 43.6 kB
JavaScript
"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 () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__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 = {
acestep: {
prettyLabel: "ACE-Step",
repoName: "ACE-Step",
repoUrl: "https://github.com/ace-step/ACE-Step",
filter: false,
countDownloads: `path:"ace_step_transformer/config.json"`,
},
"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,
},
anemoi: {
prettyLabel: "AnemoI",
repoName: "AnemoI",
repoUrl: "https://github.com/ecmwf/anemoi-inference",
docsUrl: "https://anemoi.readthedocs.io/en/latest/",
filter: false,
countDownloads: `path_extension:"ckpt"`,
snippets: snippets.anemoi,
},
araclip: {
prettyLabel: "AraClip",
repoName: "AraClip",
repoUrl: "https://huggingface.co/Arabic-Clip/araclip",
filter: false,
snippets: snippets.araclip,
},
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,
},
"bagel-mot": {
prettyLabel: "Bagel",
repoName: "Bagel",
repoUrl: "https://github.com/ByteDance-Seed/Bagel/",
filter: false,
countDownloads: `path:"llm_config.json"`,
},
bboxmaskpose: {
prettyLabel: "BBoxMaskPose",
repoName: "BBoxMaskPose",
repoUrl: "https://github.com/MiraPurkrabek/BBoxMaskPose",
filter: false,
countDownloads: `path_extension:"pth"`,
},
ben2: {
prettyLabel: "BEN2",
repoName: "BEN2",
repoUrl: "https://github.com/PramaLLC/BEN2",
snippets: snippets.ben2,
filter: false,
},
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"`,
},
birder: {
prettyLabel: "Birder",
repoName: "Birder",
repoUrl: "https://gitlab.com/birder/birder",
filter: false,
countDownloads: `path_extension:"pt"`,
},
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"`,
},
chatterbox: {
prettyLabel: "Chatterbox",
repoName: "Chatterbox",
repoUrl: "https://github.com/resemble-ai/chatterbox",
snippets: snippets.chatterbox,
countDownloads: `path:"tokenizer.json"`,
filter: false,
},
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"`,
},
comet: {
prettyLabel: "COMET",
repoName: "COMET",
repoUrl: "https://github.com/Unbabel/COMET/",
countDownloads: `path:"hparams.yaml"`,
},
contexttab: {
prettyLabel: "ConTextTab",
repoName: "ConTextTab",
repoUrl: "https://github.com/SAP-samples/contexttab",
countDownloads: `path_extension:"pt"`,
snippets: snippets.contexttab,
},
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",
repoUrl: "https://github.com/google-health/cxr-foundation",
snippets: snippets.cxr_foundation,
filter: false,
countDownloads: `path:"precomputed_embeddings/embeddings.npz" OR path:"pax-elixr-b-text/saved_model.pb"`,
},
deepforest: {
prettyLabel: "DeepForest",
repoName: "deepforest",
docsUrl: "https://deepforest.readthedocs.io/en/latest/",
repoUrl: "https://github.com/weecology/DeepForest",
},
"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,
},
"derm-foundation": {
prettyLabel: "Derm Foundation",
repoName: "derm-foundation",
repoUrl: "https://github.com/google-health/derm-foundation",
snippets: snippets.derm_foundation,
filter: false,
countDownloads: `path:"scin_dataset_precomputed_embeddings.npz" OR path:"saved_model.pb"`,
},
"describe-anything": {
prettyLabel: "Describe Anything",
repoName: "Describe Anything",
repoUrl: "https://github.com/NVlabs/describe-anything",
snippets: snippets.describe_anything,
filter: false,
},
"dia-tts": {
prettyLabel: "Dia",
repoName: "Dia",
repoUrl: "https://github.com/nari-labs/dia",
snippets: snippets.dia,
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,
},
clipscope: {
prettyLabel: "clipscope",
repoName: "clipscope",
repoUrl: "https://github.com/Lewington-pitsos/clipscope",
filter: false,
countDownloads: `path_extension:"pt"`,
},
cosyvoice: {
prettyLabel: "CosyVoice",
repoName: "CosyVoice",
repoUrl: "https://github.com/FunAudioLLM/CosyVoice",
filter: false,
countDownloads: `path_extension:"onnx" OR path_extension:"pt"`,
},
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"`,
},
fme: {
prettyLabel: "Full Model Emulation",
repoName: "Full Model Emulation",
repoUrl: "https://github.com/ai2cm/ace",
docsUrl: "https://ai2-climate-emulator.readthedocs.io/en/latest/",
filter: false,
countDownloads: `path_extension:"tar"`,
},
"gemma.cpp": {
prettyLabel: "gemma.cpp",
repoName: "gemma.cpp",
repoUrl: "https://github.com/google/gemma.cpp",
filter: false,
countDownloads: `path_extension:"sbs"`,
},
"geometry-crafter": {
prettyLabel: "GeometryCrafter",
repoName: "GeometryCrafter",
repoUrl: "https://github.com/TencentARC/GeometryCrafter",
countDownloads: `path:"point_map_vae/diffusion_pytorch_model.safetensors"`,
},
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"`,
},
hermes: {
prettyLabel: "HERMES",
repoName: "HERMES",
repoUrl: "https://github.com/LMD0311/HERMES",
filter: false,
countDownloads: `path:"ckpt/hermes_final.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"`,
},
"hunyuan3d-2": {
prettyLabel: "Hunyuan3D-2",
repoName: "Hunyuan3D-2",
repoUrl: "https://github.com/Tencent/Hunyuan3D-2",
countDownloads: `path_filename:"model_index" OR path_filename:"config"`,
},
"hunyuanworld-voyager": {
prettyLabel: "HunyuanWorld-voyager",
repoName: "HunyuanWorld-voyager",
repoUrl: "https://github.com/Tencent-Hunyuan/HunyuanWorld-Voyager",
},
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"`,
},
"index-tts": {
prettyLabel: "IndexTTS",
repoName: "IndexTTS",
repoUrl: "https://github.com/index-tts/index-tts",
snippets: snippets.indextts,
filter: false,
},
infinitetalk: {
prettyLabel: "InfiniteTalk",
repoName: "InfiniteTalk",
repoUrl: "https://github.com/MeiGen-AI/InfiniteTalk",
filter: false,
countDownloads: `path_extension:"safetensors"`,
},
"infinite-you": {
prettyLabel: "InfiniteYou",
repoName: "InfiniteYou",
repoUrl: "https://github.com/bytedance/InfiniteYou",
filter: false,
countDownloads: `path:"infu_flux_v1.0/sim_stage1/image_proj_model.bin" OR path:"infu_flux_v1.0/aes_stage2/image_proj_model.bin"`,
},
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-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,
},
"kimi-audio": {
prettyLabel: "KimiAudio",
repoName: "KimiAudio",
repoUrl: "https://github.com/MoonshotAI/Kimi-Audio",
snippets: snippets.kimi_audio,
filter: false,
},
kittentts: {
prettyLabel: "KittenTTS",
repoName: "KittenTTS",
repoUrl: "https://github.com/KittenML/KittenTTS",
snippets: snippets.kittentts,
},
kronos: {
prettyLabel: "KRONOS",
repoName: "KRONOS",
repoUrl: "https://github.com/mahmoodlab/KRONOS",
filter: false,
countDownloads: `path_extension:"pt"`,
},
k2: {
prettyLabel: "K2",
repoName: "k2",
repoUrl: "https://github.com/k2-fsa/k2",
},
"lightning-ir": {
prettyLabel: "Lightning IR",
repoName: "Lightning IR",
repoUrl: "https://github.com/webis-de/lightning-ir",
snippets: snippets.lightning_ir,
},
"litert-lm": {
prettyLabel: "LiteRT-LM",
repoName: "LiteRT-LM",
repoUrl: "https://github.com/google-ai-edge/LiteRT-LM",
filter: false,
countDownloads: `path_extension:"litertlm"`,
},
lerobot: {
prettyLabel: "LeRobot",
repoName: "LeRobot",
repoUrl: "https://github.com/huggingface/lerobot",
docsUrl: "https://huggingface.co/docs/lerobot",
filter: false,
snippets: snippets.lerobot,
},
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",
},
"magi-1": {
prettyLabel: "MAGI-1",
repoName: "MAGI-1",
repoUrl: "https://github.com/SandAI-org/MAGI-1",
countDownloads: `path:"ckpt/vae/config.json"`,
},
"magenta-realtime": {
prettyLabel: "Magenta RT",
repoName: "Magenta RT",
repoUrl: "https://github.com/magenta/magenta-realtime",
countDownloads: `path:"checkpoints/llm_base_x4286_c1860k.tar" OR path:"checkpoints/llm_large_x3047_c1860k.tar" OR path:"checkpoints/llm_large_x3047_c1860k/checkpoint"`,
},
"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,
},
matanyone: {
prettyLabel: "MatAnyone",
repoName: "MatAnyone",
repoUrl: "https://github.com/pq-yang/MatAnyone",
snippets: snippets.matanyone,
filter: false,
},
"mesh-anything": {
prettyLabel: "MeshAnything",
repoName: "MeshAnything",
repoUrl: "https://github.com/buaacyw/MeshAnything",
filter: false,
countDownloads: `path:"MeshAnything_350m.pth"`,
snippets: snippets.mesh_anything,
},
merlin: {
prettyLabel: "Merlin",
repoName: "Merlin",
repoUrl: "https://github.com/StanfordMIMI/Merlin",
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",
repoUrl: "https://github.com/mit-nlp/MITIE",
countDownloads: `path_filename:"total_word_feature_extractor"`,
},
"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"`,
},
mtvcraft: {
prettyLabel: "MTVCraft",
repoName: "MTVCraft",
repoUrl: "https://github.com/baaivision/MTVCraft",
filter: false,
countDownloads: `path:"vae/3d-vae.pt"`,
},
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" OR path_extension:"json"`,
},
"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:"open_clip_model.safetensors"
OR path:"model.safetensors"
OR path:"open_clip_pytorch_model.bin"
OR path:"pytorch_model.bin"`,
},
openpeerllm: {
prettyLabel: "OpenPeerLLM",
repoName: "OpenPeerLLM",
repoUrl: "https://huggingface.co/openpeerai/openpeerllm",
docsUrl: "https://huggingface.co/OpenPeerAI/OpenPeerLLM/blob/main/README.md",
countDownloads: `path:".meta-huggingface.json"`,
filter: false,
},
"open-sora": {
prettyLabel: "Open-Sora",
repoName: "Open-Sora",
repoUrl: "https://github.com/hpcaitech/Open-Sora",
filter: false,
countDownloads: `path:"Open_Sora_v2.safetensors"`,
},
outetts: {
prettyLabel: "OuteTTS",
repoName: "OuteTTS",
repoUrl: "https://github.com/edwko/OuteTTS",
snippets: snippets.outetts,
filter: false,
},
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"`,
},
PaddleOCR: {
prettyLabel: "PaddleOCR",
repoName: "PaddleOCR",
repoUrl: "https://github.com/PaddlePaddle/PaddleOCR",
snippets: snippets.paddleocr,
filter: true,
},
peft: {
prettyLabel: "PEFT",
repoName: "PEFT",
repoUrl: "https://github.com/huggingface/peft",
snippets: snippets.peft,
filter: true,
countDownloads: `path:"adapter_config.json"`,
},
"perception-encoder": {
prettyLabel: "PerceptionEncoder",
repoName: "PerceptionModels",
repoUrl: "https://github.com/facebookresearch/perception_models",
filter: false,
snippets: snippets.perception_encoder,
countDownloads: `path_extension:"pt"`,
},
"phantom-wan": {
prettyLabel: "Phantom",
repoName: "Phantom",
repoUrl: "https://github.com/Phantom-video/Phantom",
snippets: snippets.phantom_wan,
filter: false,
countDownloads: `path_extension:"pth"`,
},
"pruna-ai": {
prettyLabel: "Pruna AI",
repoName: "Pruna AI",
repoUrl: "https://github.com/PrunaAI/pruna",
snippets: snippets.pruna,
docsUrl: "https://docs.pruna.ai",
},
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"`,
},
renderformer: {
prettyLabel: "RenderFormer",
repoName: "RenderFormer",
repoUrl: "https://github.com/microsoft/renderformer",
snippets: snippets.renderformer,
filter: false,
},
reverb: {
prettyLabel: "Reverb",
repoName: "Reverb",
repoUrl: "https://github.com/revdotcom/reverb",
filter: false,
},
rkllm: {
prettyLabel: "RKLLM",
repoName: "RKLLM",
repoUrl: "https://github.com/airockchip/rknn-llm",
countDownloads: `path_extension:"rkllm"`,
},
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"`,
},
seedvr: {
prettyLabel: "SeedVR",
repoName: "SeedVR",
repoUrl: "https://github.com/ByteDance-Seed/SeedVR",
filter: false,
countDownloads: `path_extension:"pth"`,
},
"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,
},
monkeyocr: {
prettyLabel: "MonkeyOCR",
repoName: "monkeyocr",
repoUrl: "https://github.com/Yuliang-Liu/MonkeyOCR",
filter: false,
countDownloads: `path:"Recognition/config.json"`,
},
"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"`,
},
songbloom: {
prettyLabel: "SongBloom",
repoName: "SongBloom",
repoUrl: "https://github.com/Cypress-Yang/SongBloom",
filter: false,
countDownloads: `path_extension:"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"`,
},
swarmformer: {
prettyLabel: "SwarmFormer",
repoName: "SwarmFormer",
repoUrl: "https://github.com/takara-ai/SwarmFormer",
snippets: snippets.swarmformer,
filter: false,
},
"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"`,
},
"tencent-song-generation": {
prettyLabel: "SongGeneration",
repoName: "SongGeneration",
repoUrl: "https://github.com/tencent-ailab/songgeneration",
filter: false,
countDownloads: `path:"ckpt/songgeneration_base/model.pt"`,
},
tensorflowtts: {
prettyLabel: "TensorFlowTTS",
repoName: "TensorFlowTTS",
repoUrl: "https://github.com/TensorSpeech/TensorFlowTTS",
snippets: snippets.tensorflowtts,
},
tabpfn: {
prettyLabel: "TabPFN",
repoName: "TabPFN",
repoUrl: "https://github.com/PriorLabs/TabPFN",
},
terratorch: {
prettyLabel: "TerraTorch",
repoName: "TerraTorch",
repoUrl: "https://github.com/IBM/terratorch",
docsUrl: "https://ibm.github.io/terratorch/",
filter: false,
countDownloads: `path_extension:"pt" OR path_extension:"ckpt"`,
snippets: snippets.terratorch,
},
"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" OR path:"checkpoints/checkpoint_2150000/state/checkpoint" OR path_extension:"ckpt"`,
},
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"`,
},
tirex: {
prettyLabel: "TiRex",
repoName: "TiRex",
repoUrl: "https://github.com/NX-AI/tirex",
countDownloads: `path_extension:"ckpt"`,
},
torchgeo: {
prettyLabel: "TorchGeo",
repoName: "TorchGeo",
repoUrl: "https://github.com/microsoft/torchgeo",
docsUrl: "https://torchgeo.readthedocs.io/",
filter: false,
countDownloads: `path_extension:"pt" OR path_extension:"pth"`,
},
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,
},
trellis: {
prettyLabel: "Trellis",
repoName: "Trellis",
repoUrl: "https://github.com/microsoft/TRELLIS",
countDownloads: `path_extension:"safetensors"`,
},
ultralytics: {
prettyLabel: "ultralytics",
repoName: "ultralytics",
repoUrl: "https://github.com/ultralytics/ultralytics",
docsUrl: "https://github.com/ultralytics/ultralytics",
filter: false,
countDownloads: `path_extension:"pt"`,
snippets: snippets.ultralytics,
},
univa: {
prettyLabel: "univa",
repoName: "univa",
repoUrl: "https://github.com/PKU-YuanGroup/UniWorld-V1",
snippets: snippets.univa,
filter: true,
countDownloads: `path:"config.json"`,
},
"uni-3dar": {
prettyLabel: "Uni-3DAR",
repoName: "Uni-3DAR",
repoUrl: "https://github.com/dptech-corp/Uni-3DAR",
docsUrl: "https://github.com/dptech-corp/Uni-3DAR",
countDownloads: `path_extension:"pt"`,
},
"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"`,
},
sana: {
prettyLabel: "Sana",
repoName: "Sana",
repoUrl: "https://github.com/NVlabs/Sana",
countDownloads: `path_extension:"pth"`,
snippets: snippets.sana,
},
videoprism: {
prettyLabel: "VideoPrism",
repoName: "VideoPrism",
repoUrl: "https://github.com/google-deepmind/videoprism",
countDownloads: `path_extension:"npz"`,
snippets: snippets.videoprism,
},
"vfi-mamba": {
prettyLabel: "VFIMamba",
repoName: "VFIMamba",
repoUrl: "https://github.com/MCG-NJU/VFIMamba",
countDownloads: `path_extension:"pkl"`,
snippets: snippets.vfimamba,
},
lvface: {
prettyLabel: "LVFace",
repoName: "LVFace",
repoUrl: "https://github.com/bytedance/LVFace",
countDownloads: `path_extension:"pt" OR path_extension:"onnx"`,
snippets: snippets.lvface,
},
voicecraft: {
prettyLabel: "VoiceCraft",
repoName: "VoiceCraft",
repoUrl: "https://github.com/jasonppy/VoiceCraft",
docsUrl: "https://github.com/jasonppy/VoiceCraft",
snippets: snippets.voicecraft,
},
vui: {
prettyLabel: "Vui",
repoName: "Vui",
repoUrl: "https://github.com/vui-ai/vui",
countDownloads: `path_extension:"pt"`,
snippets: snippets.vui,
},
"wan2.2": {
prettyLabel: "Wan2.2",
repoName: "Wan2.2",
repoUrl: "https://github.com/Wan-Video/Wan2.2",
countDownloads: `path_filename:"config" AND path_extension:"json"`,
},
wham: {
prettyLabel: "WHAM",
repoName: "wham",
repoUrl: "https://huggingface.co/microsoft/wham",
docsUrl: "https://huggingface.co/microsoft/wham/blob/main/README.md",
countDownloads: `path_extension:"ckpt"`,
},
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"`,
},
yolov10: {
// YOLOv10 is a fork of ultraLytics. Code snippets and download count are the same but the repo is different.
prettyLabel: "YOLOv10",
repoName: "YOLOv10",
repoUrl: "https://github.com/THU-MIG/yolov10",
docsUrl: "https://github.com/THU-MIG/yolov10",
countDownloads: `path_extension:"pt" OR path_extension:"safetensors"`,
snippets: snippets.ultralytics,
},
zonos: {
prettyLabel: "Zonos",
repoName: "Zonos",
repoUrl: "https://github.com/Zyphra/Zonos",
docsUrl: "https://github.com/Zyphra/Zonos",
snippets: snippets.zonos,
filter: false,
},
"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);