Spaces:
Sleeping
Sleeping
| import * as snippets from "./model-libraries-snippets.js"; | |
| import type { ModelData } from "./model-data.js"; | |
| import type { ElasticSearchQuery } from "./model-libraries-downloads.js"; | |
| /** | |
| * Elements configurable by a model library. | |
| */ | |
| export interface LibraryUiElement { | |
| /** | |
| * Pretty name of the library. | |
| * displayed in tags, and on the main | |
| * call-to-action button on the model page. | |
| */ | |
| prettyLabel: string; | |
| /** | |
| * Repo name of the library's (usually on GitHub) code repo | |
| */ | |
| repoName: string; | |
| /** | |
| * URL to library's (usually on GitHub) code repo | |
| */ | |
| repoUrl: string; | |
| /** | |
| * URL to library's docs | |
| */ | |
| docsUrl?: string; | |
| /** | |
| * Code snippet(s) displayed on model page | |
| */ | |
| snippets?: (model: ModelData) => string[]; | |
| /** | |
| * Elastic query used to count this library's model downloads | |
| * | |
| * By default, those files are counted: | |
| * "config.json", "config.yaml", "hyperparams.yaml", "params.json", "meta.yaml" | |
| */ | |
| countDownloads?: ElasticSearchQuery; | |
| /** | |
| * should we display this library in hf.co/models filter | |
| * (only for popular libraries with > 100 models) | |
| */ | |
| filter?: boolean; | |
| } | |
| /** | |
| * 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. | |
| */ | |
| export const 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, | |
| }, | |
| "aviation-ner": { | |
| prettyLabel: "Aviation NER", | |
| repoName: "Aviation NER", | |
| repoUrl: "https://github.com/Boeing/aviation_ner_sdr", | |
| docsUrl: "https://github.com/Boeing/aviation_ner_sdr", | |
| countDownloads: `path:"gliner_config.json"`, | |
| filter: false, | |
| }, | |
| 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"`, | |
| }, | |
| boltzgen: { | |
| prettyLabel: "BoltzGen", | |
| repoName: "BoltzGen", | |
| repoUrl: "https://github.com/HannesStark/boltzgen", | |
| filter: false, | |
| countDownloads: `path:"boltzgen1_diverse.ckpt"`, | |
| }, | |
| cancertathomev2: { | |
| prettyLabel: "Cancer@HomeV2", | |
| repoName: "Cancer@HomeV2", | |
| repoUrl: "https://huggingface.co/OpenPeerAI/CancerAtHomeV2", | |
| filter: false, | |
| countDownloads: `path:"run.py"`, | |
| }, | |
| 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, | |
| }, | |
| 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, | |
| }, | |
| chaossim: { | |
| prettyLabel: "ChaosSIM", | |
| repoName: "ChaosSIM", | |
| repoUrl: "https://huggingface.co/OpenPeerAI/ChaosSIM/", | |
| countDownloads: `path:"ChaosSim.nb"`, | |
| 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"`, | |
| }, | |
| "chronos-forecasting": { | |
| prettyLabel: "Chronos", | |
| repoName: "Chronos", | |
| repoUrl: "https://github.com/amazon-science/chronos-forecasting", | |
| snippets: snippets.chronos_forecasting, | |
| }, | |
| clara: { | |
| prettyLabel: "Clara", | |
| repoName: "Clara", | |
| filter: false, | |
| repoUrl: "https://github.com/nvidia/clara", | |
| countDownloads: `path_extension:"ckpt" OR path:"config.json"`, | |
| }, | |
| clipscope: { | |
| prettyLabel: "clipscope", | |
| repoName: "clipscope", | |
| repoUrl: "https://github.com/Lewington-pitsos/clipscope", | |
| filter: false, | |
| countDownloads: `path_extension:"pt"`, | |
| }, | |
| "cloud-agents": { | |
| prettyLabel: "Cloud Agents", | |
| repoName: "Cloud Agents", | |
| repoUrl: "https://huggingface.co/OpenPeerAI/Cloud-Agents", | |
| filter: false, | |
| countDownloads: `path:"setup.py"`, | |
| }, | |
| colipri: { | |
| prettyLabel: "COLIPRI", | |
| repoName: "COLIPRI", | |
| repoUrl: "https://huggingface.co/microsoft/colipri", | |
| snippets: snippets.colipri, | |
| filter: false, | |
| countDownloads: `path_extension:"safetensors"`, | |
| }, | |
| 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"`, | |
| }, | |
| 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"`, | |
| }, | |
| 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, | |
| }, | |
| dia2: { | |
| prettyLabel: "Dia2", | |
| repoName: "Dia2", | |
| repoUrl: "https://github.com/nari-labs/dia2", | |
| snippets: snippets.dia2, | |
| filter: false, | |
| }, | |
| "diff-interpretation-tuning": { | |
| prettyLabel: "Diff Interpretation Tuning", | |
| repoName: "Diff Interpretation Tuning", | |
| repoUrl: "https://github.com/Aviously/diff-interpretation-tuning", | |
| filter: false, | |
| countDownloads: `path_extension:"pt"`, | |
| }, | |
| 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, | |
| }, | |
| "docking-at-home": { | |
| prettyLabel: "Docking@Home", | |
| repoName: "Docking@Home", | |
| repoUrl: "https://huggingface.co/OpenPeerAI/DockingAtHOME", | |
| filter: false, | |
| countDownloads: `path:"setup.py"`, | |
| }, | |
| doctr: { | |
| prettyLabel: "docTR", | |
| repoName: "doctr", | |
| repoUrl: "https://github.com/mindee/doctr", | |
| }, | |
| 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, | |
| }, | |
| fastprint: { | |
| prettyLabel: "Fast Print", | |
| repoName: "Fast Print", | |
| repoUrl: "https://huggingface.co/OpenPeerAI/FastPrint", | |
| countDownloads: `path_extension:"cs"`, | |
| }, | |
| fasttext: { | |
| prettyLabel: "fastText", | |
| repoName: "fastText", | |
| repoUrl: "https://fasttext.cc/", | |
| snippets: snippets.fasttext, | |
| filter: true, | |
| countDownloads: `path_extension:"bin"`, | |
| }, | |
| fixer: { | |
| prettyLabel: "Fixer", | |
| repoName: "Fixer", | |
| repoUrl: "https://github.com/nv-tlabs/Fixer", | |
| filter: false, | |
| countDownloads: `path:"pretrained/pretrained_fixer.pkl"`, | |
| }, | |
| 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"`, | |
| }, | |
| gliner2: { | |
| prettyLabel: "GLiNER2", | |
| repoName: "GLiNER2", | |
| repoUrl: "https://github.com/fastino-ai/GLiNER2", | |
| snippets: snippets.gliner2, | |
| filter: false, | |
| }, | |
| "glm-tts": { | |
| prettyLabel: "GLM-TTS", | |
| repoName: "GLM-TTS", | |
| repoUrl: "https://github.com/zai-org/GLM-TTS", | |
| filter: false, | |
| countDownloads: `path:"flow/flow.pt"`, | |
| }, | |
| "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"`, | |
| }, | |
| "habibi-tts": { | |
| prettyLabel: "Habibi-TTS", | |
| repoName: "Habibi-TTS", | |
| repoUrl: "https://github.com/SWivid/Habibi-TTS", | |
| filter: false, | |
| countDownloads: `path_extension:"safetensors"`, | |
| }, | |
| 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", | |
| }, | |
| "hy-worldplay": { | |
| prettyLabel: "HY-WorldPlay", | |
| repoName: "HY-WorldPlay", | |
| repoUrl: "https://github.com/Tencent-Hunyuan/HY-WorldPlay", | |
| filter: false, | |
| countDownloads: `path_extension:"json"`, | |
| }, | |
| "image-matching-models": { | |
| prettyLabel: "Image Matching Models", | |
| repoName: "Image Matching Models", | |
| repoUrl: "https://github.com/alexstoken/image-matching-models", | |
| filter: false, | |
| countDownloads: `path_extension:"safetensors"`, | |
| }, | |
| 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"`, | |
| }, | |
| intellifold: { | |
| prettyLabel: "IntelliFold", | |
| repoName: "IntelliFold", | |
| repoUrl: "https://github.com/IntelliGen-AI/IntelliFold", | |
| filter: false, | |
| countDownloads: `path_extension:"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-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, | |
| }, | |
| kernels: { | |
| prettyLabel: "Kernels", | |
| repoName: "Kernels", | |
| repoUrl: "https://github.com/huggingface/kernels", | |
| docsUrl: "https://huggingface.co/docs/kernels", | |
| snippets: snippets.kernels, | |
| countDownloads: `path_filename:"_ops" AND path_extension:"py"`, | |
| }, | |
| "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: { | |
| prettyLabel: "LiteRT", | |
| repoName: "LiteRT", | |
| repoUrl: "https://github.com/google-ai-edge/LiteRT", | |
| filter: false, | |
| countDownloads: `path_extension:"tflite"`, | |
| }, | |
| "litert-lm": { | |
| prettyLabel: "LiteRT-LM", | |
| repoName: "LiteRT-LM", | |
| repoUrl: "https://github.com/google-ai-edge/LiteRT-LM", | |
| filter: false, | |
| countDownloads: `path_extension:"litertlm" OR path_extension:"task"`, | |
| }, | |
| lerobot: { | |
| prettyLabel: "LeRobot", | |
| repoName: "LeRobot", | |
| repoUrl: "https://github.com/huggingface/lerobot", | |
| docsUrl: "https://huggingface.co/docs/lerobot", | |
| filter: false, | |
| snippets: snippets.lerobot, | |
| }, | |
| lightglue: { | |
| prettyLabel: "LightGlue", | |
| repoName: "LightGlue", | |
| repoUrl: "https://github.com/cvg/LightGlue", | |
| filter: false, | |
| countDownloads: `path_extension:"pth" OR path:"config.json"`, | |
| }, | |
| 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, | |
| }, | |
| "manas-1": { | |
| prettyLabel: "MANAS-1", | |
| repoName: "MANAS-1", | |
| repoUrl: "https://github.com/NeurodxAI/manas-1", | |
| countDownloads: `path_extension:"pt"`, | |
| }, | |
| "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"`, | |
| }, | |
| "ml-sharp": { | |
| prettyLabel: "Sharp", | |
| repoName: "Sharp", | |
| repoUrl: "https://github.com/apple/ml-sharp", | |
| filter: false, | |
| countDownloads: `path_extension:"pt"`, | |
| }, | |
| 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", | |
| snippets: snippets.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", | |
| docsUrl: "https://www.paddleocr.ai/", | |
| snippets: snippets.paddleocr, | |
| filter: true, | |
| countDownloads: `path_extension:"safetensors" OR path:"inference.pdiparams"`, | |
| }, | |
| 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"`, | |
| }, | |
| "pocket-tts": { | |
| prettyLabel: "Pocket-TTS", | |
| repoName: "PocketTTS", | |
| repoUrl: "https://github.com/kyutai-labs/pocket-tts", | |
| snippets: snippets.pocket_tts, | |
| filter: false, | |
| countDownloads: `path:"tts_b6369a24.safetensors"`, | |
| }, | |
| "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, | |
| }, | |
| quantumpeer: { | |
| prettyLabel: "QuantumPeer", | |
| repoName: "QuantumPeer", | |
| repoUrl: "https://github.com/OpenPeer-AI/QuantumPeer", | |
| filter: false, | |
| countDownloads: `path_extension:"setup.py"`, | |
| }, | |
| qwen3_tts: { | |
| prettyLabel: "Qwen3-TTS", | |
| repoName: "Qwen3-TTS", | |
| repoUrl: "https://github.com/QwenLM/Qwen3-TTS", | |
| snippets: snippets.qwen3_tts, | |
| 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"`, | |
| }, | |
| "sam-3d-body": { | |
| prettyLabel: "SAM 3D Body", | |
| repoName: "SAM 3D Body", | |
| repoUrl: "https://github.com/facebookresearch/sam-3d-body", | |
| filter: false, | |
| snippets: snippets.sam_3d_body, | |
| countDownloads: `path:"model_config.yaml"`, | |
| }, | |
| "sam-3d-objects": { | |
| prettyLabel: "SAM 3D Objects", | |
| repoName: "SAM 3D Objects", | |
| repoUrl: "https://github.com/facebookresearch/sam-3d-objects", | |
| filter: false, | |
| snippets: snippets.sam_3d_objects, | |
| countDownloads: `path:"checkpoints/pipeline.yaml"`, | |
| }, | |
| same: { | |
| prettyLabel: "SAME", | |
| repoName: "SAME", | |
| repoUrl: "https://github.com/GengzeZhou/SAME", | |
| filter: false, | |
| countDownloads: `path:"ckpt/SAME.pt" OR path:"pretrain/Attnq_pretrained_ckpt.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"`, | |
| }, | |
| "sap-rpt-1-oss": { | |
| prettyLabel: "sap-rpt-1-oss", | |
| repoName: "sap-rpt-1-oss", | |
| repoUrl: "https://github.com/SAP-samples/sap-rpt-1-oss", | |
| countDownloads: `path_extension:"pt"`, | |
| snippets: snippets.sap_rpt_one_oss, | |
| }, | |
| 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"`, | |
| }, | |
| "self-forcing": { | |
| prettyLabel: "SelfForcing", | |
| repoName: "SelfForcing", | |
| repoUrl: "https://github.com/guandeh17/Self-Forcing", | |
| filter: false, | |
| countDownloads: `path_extension:"pt"`, | |
| }, | |
| "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"`, | |
| }, | |
| supertonic: { | |
| prettyLabel: "Supertonic", | |
| repoName: "Supertonic", | |
| repoUrl: "https://github.com/supertone-inc/supertonic", | |
| snippets: snippets.supertonic, | |
| filter: false, | |
| }, | |
| 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, | |
| }, | |
| tensorrt: { | |
| prettyLabel: "TensorRT", | |
| repoName: "TensorRT", | |
| repoUrl: "https://github.com/NVIDIA/TensorRT", | |
| countDownloads: `path_extension:"onnx"`, | |
| }, | |
| 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, | |
| }, | |
| vismatch: { | |
| prettyLabel: "VisMatch", | |
| repoName: "VisMatch", | |
| repoUrl: "https://github.com/gmberton/vismatch", | |
| filter: false, | |
| countDownloads: `path:"vismatch.yaml"`, | |
| }, | |
| 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, | |
| }, | |
| voxcpm: { | |
| prettyLabel: "VoxCPM", | |
| repoName: "VoxCPM", | |
| repoUrl: "https://github.com/OpenBMB/VoxCPM", | |
| snippets: snippets.voxcpm, | |
| filter: false, | |
| }, | |
| vui: { | |
| prettyLabel: "Vui", | |
| repoName: "Vui", | |
| repoUrl: "https://github.com/vui-ai/vui", | |
| countDownloads: `path_extension:"pt"`, | |
| snippets: snippets.vui, | |
| }, | |
| vibevoice: { | |
| prettyLabel: "VibeVoice", | |
| repoName: "VibeVoice", | |
| repoUrl: "https://github.com/microsoft/VibeVoice", | |
| snippets: snippets.vibevoice, | |
| filter: false, | |
| }, | |
| videox_fun: { | |
| prettyLabel: "VideoX Fun", | |
| repoName: "VideoX Fun", | |
| repoUrl: "https://github.com/aigc-apps/VideoX-Fun", | |
| filter: false, | |
| countDownloads: `path_extension:"safetensors"`, | |
| }, | |
| "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, | |
| }, | |
| yolov26: { | |
| prettyLabel: "YOLOv26", | |
| repoName: "YOLOv26", | |
| repoUrl: "https://github.com/ultralytics/ultralytics", | |
| docsUrl: "https://docs.ultralytics.com/models/yolo26/", | |
| countDownloads: `path_extension:"pt" OR path_extension:"safetensors"`, | |
| }, | |
| 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, | |
| }, | |
| } satisfies Record<string, LibraryUiElement>; | |
| export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS; | |
| export const ALL_MODEL_LIBRARY_KEYS = Object.keys(MODEL_LIBRARIES_UI_ELEMENTS) as ModelLibraryKey[]; | |
| export const ALL_DISPLAY_MODEL_LIBRARY_KEYS = ( | |
| Object.entries(MODEL_LIBRARIES_UI_ELEMENTS as Record<ModelLibraryKey, LibraryUiElement>) as [ | |
| ModelLibraryKey, | |
| LibraryUiElement, | |
| ][] | |
| ) | |
| // eslint-disable-next-line @typescript-eslint/no-unused-vars | |
| .filter(([_, v]) => v.filter) | |
| .map(([k]) => k); | |