Models: Difference between revisions

Revision as of 19:46, 2 February 2023

See also: Terms, Applications and Guides

Computer Vision Models

See also: Computer Vision Models

Model	Name	User/Org	Task	Library	Dataset	Paper	Related to	License
CIDAS/clipseg-rd64-refined model	clipseg-rd64-refined	http://CIDAS	Image Segmentation	PyTorch Transformers		2112.10003	Vision Clipseg	Apache-2.0
CompVis/stable-diffusion-safety-checker model	stable-diffusion-safety-checker	http://CompVis		PyTorch Transformers		1910.09700 2103.00020	Clip
Facebook/convnext-base-224 model	convnext-base-224	http://facebook	Image Classification	PyTorch TensorFlow Transformers	Imagenet-1k	2201.03545	Vision AutoTrain Compatible Convnext	Apache-2.0
Facebook/convnext-large-224 model	convnext-large-224	http://facebook	Image Classification	PyTorch TensorFlow Transformers	Imagenet-1k	2201.03545	Vision AutoTrain Compatible Convnext	Apache-2.0
Google/vit-base-patch16-224 model	vit-base-patch16-224	http://google	Image Classification	PyTorch TensorFlow JAX Transformers	Imagenet-21k Imagenet-1k	2010.11929 2006.03677	Vision Vit AutoTrain Compatible	Apache-2.0
Microsoft/resnet-18 model	resnet-18	http://microsoft	Image Classification	PyTorch TensorFlow Safetensors Transformers	Imagenet-1k	1512.03385	Vision AutoTrain Compatible Resnet	Apache-2.0
Microsoft/resnet-50 model	resnet-50	http://microsoft	Image Classification	PyTorch TensorFlow JAX Transformers	Imagenet-1k	1512.03385	Vision AutoTrain Compatible Resnet	Apache-2.0
Openai/clip-vit-base-patch16 model	clip-vit-base-patch16	http://openai	Zero-Shot Image Classification	PyTorch JAX Transformers		2103.00020 1908.04913	Clip Vision
Openai/clip-vit-base-patch32 model	clip-vit-base-patch32	http://openai	Zero-Shot Image Classification	PyTorch TensorFlow JAX Transformers		2103.00020 1908.04913	Clip Vision
Openai/clip-vit-large-patch14 model	clip-vit-large-patch14	http://openai	Zero-Shot Image Classification	PyTorch TensorFlow JAX Transformers		2103.00020 1908.04913	Clip Vision

Natural Language Processing Models

See also: Natural Language Processing Models

Model	Name	User/Org	Task	Library	Dataset	Language	Paper	Related to	License
Albert-base-v2 model	albert-base-v2		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers Rust	Wikipedia Bookcorpus	English	1909.11942	Albert AutoTrain Compatible	Apache-2.0
Allenai/longformer-base-4096 model	longformer-base-4096	http://allenai		PyTorch TensorFlow Transformers Rust		English	2004.05150	Longformer	Apache-2.0
Allenai/scibert scivocab uncased model	scibert_scivocab_uncased	http://allenai		PyTorch JAX Transformers		English		BERT
Bert-base-cased model	bert-base-cased		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers	Wikipedia Bookcorpus	English	1810.04805	Exbert BERT AutoTrain Compatible	Apache-2.0
Bert-base-chinese model	bert-base-chinese		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers		Chinese	1810.04805	BERT AutoTrain Compatible
Bert-base-multilingual-cased model	bert-base-multilingual-cased		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers	Wikipedia	104 languages	1810.04805	BERT AutoTrain Compatible	Apache-2.0
Bert-base-uncased model	bert-base-uncased		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers Rust Core ML	Wikipedia Bookcorpus	English	1810.04805	Exbert BERT AutoTrain Compatible	Apache-2.0
Bigcode/santacoder model	santacoder	http://bigcode	Text Generation	PyTorch Transformers	Bigcode/the-stack	Code	1911.02150 2207.14255 2301.03988	Gpt2 Eval Results	Bigcode-openrail-m
Bigscience/bloom-560m model	bloom-560m	http://bigscience	Text Generation	PyTorch JAX Safetensors Transformers		48 languages	1909.08053 2110.02861 2108.12409	Bloom	Bigscience-bloom-rail-1.0
Camembert-base model	camembert-base		Fill-Mask	PyTorch TensorFlow Safetensors Transformers	Oscar	French	1911.03894	Camembert AutoTrain Compatible	Mit
Cardiffnlp/twitter-roberta-base-sentiment model	twitter-roberta-base-sentiment	http://cardiffnlp	Text Classification	PyTorch TensorFlow JAX Transformers	Tweet eval	English	2010.12421	Roberta
Cardiffnlp/twitter-roberta-base-sentiment-latest model	twitter-roberta-base-sentiment-latest	http://cardiffnlp	Text Classification	PyTorch TensorFlow Transformers	Tweet eval	English	2202.03829	Roberta
Deepset/roberta-base-squad2 model	roberta-base-squad2	http://deepset	Question Answering	PyTorch TensorFlow JAX Safetensors Transformers Rust	Squad v2	English		Roberta Eval Results AutoTrain Compatible	Cc-by-4.0
Deepset/sentence bert model	sentence_bert	http://deepset		PyTorch JAX Transformers				BERT	Apache-2.0
Distilbert-base-multilingual-cased model	distilbert-base-multilingual-cased		Fill-Mask	PyTorch TensorFlow Safetensors Transformers ONNX	Wikipedia	104 languages	1910.09700 1910.01108	Distilbert AutoTrain Compatible	Apache-2.0
Distilbert-base-uncased model	distilbert-base-uncased		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers Rust	Wikipedia Bookcorpus	English	1910.01108	Exbert Distilbert AutoTrain Compatible	Apache-2.0
Distilbert-base-uncased-finetuned-sst-2-english model	distilbert-base-uncased-finetuned-sst-2-english		Text Classification	PyTorch TensorFlow Safetensors Transformers Rust	Sst2 Glue	English	1910.01108	Distilbert Eval Results	Apache-2.0
Distilgpt2 model	distilgpt2		Text Generation	PyTorch TensorFlow JAX Safetensors Transformers Rust TF Lite Core ML	Openwebtext	English	1910.09700 1910.01108 2201.08542 2203.12574 1503.02531	Gpt2 Exbert Eval Results Carbon Emissions	Apache-2.0
Distilroberta-base model	distilroberta-base		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers Rust	Openwebtext	English	1910.09700 1910.01108	Exbert Roberta AutoTrain Compatible	Apache-2.0
Dslim/bert-base-NER model	bert-base-NER	http://dslim	Token Classification	PyTorch TensorFlow JAX Safetensors Transformers	Conll2003	English	1810.04805	BERT AutoTrain Compatible	Mit
Facebook/bart-large-cnn model	bart-large-cnn	http://facebook	Text2Text Generation Summarization	PyTorch TensorFlow JAX Transformers Rust	Cnn dailymail	English	1910.13461	Bart Eval Results AutoTrain Compatible	Mit
Facebook/bart-large-mnli model	bart-large-mnli	http://facebook	Text Classification Zero-Shot Classification	PyTorch JAX Safetensors Transformers Rust	Multi nli		1910.13461 1909.00161	Bart	Mit
Google/bert uncased L-2 H-128 A-2 model	bert_uncased_L-2_H-128_A-2	http://google		PyTorch JAX Transformers			1908.08962	BERT	Apache-2.0
Google/electra-base-discriminator model	electra-base-discriminator	http://google		PyTorch TensorFlow JAX Transformers Rust		English		Electra Pretraining	Apache-2.0
Gpt2 model	gpt2		Text Generation	PyTorch TensorFlow JAX Safetensors Transformers Rust TF Lite		English		Gpt2 Exbert	Mit
Gpt2-large model	gpt2-large		Text Generation	PyTorch TensorFlow JAX Safetensors Transformers Rust		English	1910.09700	Gpt2	Mit
Helsinki-NLP/opus-mt-en-es model	opus-mt-en-es	http://Helsinki-NLP	Text2Text Generation Translation	PyTorch TensorFlow JAX Transformers		English Spanish		Marian AutoTrain Compatible	Apache-2.0
Lengyue233/content-vec-best model	content-vec-best	http://lengyue233		PyTorch Transformers				Hubert	Mit
Microsoft/deberta-base model	deberta-base	http://microsoft	Fill-Mask	PyTorch TensorFlow Transformers Rust		English	2006.03654	Deberta Deberta-v1	Mit
Microsoft/layoutlm-base-uncased model	layoutlm-base-uncased	http://microsoft		PyTorch TensorFlow Transformers			1912.13318	Layoutlm
Microsoft/layoutlmv3-base model	layoutlmv3-base	http://microsoft		PyTorch TensorFlow Transformers ONNX		English	2204.08387	Layoutlmv3	Cc-by-nc-sa-4.0
Prajjwal1/bert-tiny model	bert-tiny	http://prajjwal1		PyTorch Transformers		English	1908.08962 2110.01518	BERT MNLI NLI Transformer Pre-training	Mit
Roberta-base model	roberta-base		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers Rust	Wikipedia Bookcorpus	English	1907.11692 1806.02847	Exbert Roberta AutoTrain Compatible	Mit
Roberta-large model	roberta-large		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers ONNX	Wikipedia Bookcorpus	English	1907.11692 1806.02847	Exbert Roberta AutoTrain Compatible	Mit
SZTAKI-HLT/hubert-base-cc model	hubert-base-cc	http://SZTAKI-HLT		PyTorch TensorFlow JAX Transformers	Wikipedia Common crawl	Hungarian		BERT	Apache-2.0
Sentence-transformers/all-MiniLM-L6-v2 model	all-MiniLM-L6-v2	http://sentence-transformers	Sentence Similarity Feature Extraction	PyTorch TensorFlow Rust Sentence Transformers	S2orc Flax-sentence-embeddings/stackexchange xml Ms marco Gooaq Yahoo answers topics Code search net Search qa Eli5 Snli Multi nli Wikihow Natural questions Trivia qa Embedding-data/sentence-compression Embedding-data/flickr30k-captions Embedding-data/altlex Embedding-data/simple-wiki Embedding-data/QQP Embedding-data/SPECTER Embedding-data/PAQ pairs Embedding-data/WikiAnswers	English	1904.06472 2102.07033 2104.08727 1704.05179 1810.09305	BERT	Apache-2.0
Sentence-transformers/all-mpnet-base-v2 model	all-mpnet-base-v2	http://sentence-transformers	Sentence Similarity Feature Extraction	PyTorch Sentence Transformers	S2orc Flax-sentence-embeddings/stackexchange xml Gooaq Yahoo answers topics Code search net Search qa Eli5 Snli Multi nli Wikihow Natural questions Trivia qa Embedding-data/sentence-compression Embedding-data/flickr30k-captions Embedding-data/altlex Embedding-data/simple-wiki Embedding-data/QQP Embedding-data/SPECTER Embedding-data/PAQ pairs Embedding-data/WikiAnswers MS Marco	English	1904.06472 2102.07033 2104.08727 1704.05179 1810.09305	Mpnet	Apache-2.0
T5-base model	t5-base		Text2Text Generation Translation Summarization	PyTorch TensorFlow JAX Safetensors Transformers Rust	C4	4 languages	1805.12471 1708.00055 1704.05426 1606.05250 1808.09121 1810.12885 1905.10044 1910.09700	T5 AutoTrain Compatible	Apache-2.0
T5-small model	t5-small		Text2Text Generation Translation Summarization	PyTorch TensorFlow JAX Safetensors Transformers Rust	C4	5 languages	1805.12471 1708.00055 1704.05426 1606.05250 1808.09121 1810.12885 1905.10044 1910.09700	T5 AutoTrain Compatible	Apache-2.0
Xlm-roberta-base model	xlm-roberta-base		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers ONNX		94 languages	1911.02116	Exbert Xlm-roberta AutoTrain Compatible	Mit
Xlm-roberta-large model	xlm-roberta-large		Fill-Mask	PyTorch TensorFlow JAX Safetensors Transformers ONNX		94 languages	1911.02116	Exbert Xlm-roberta AutoTrain Compatible	Mit

Audio Models

See also: Audio Models

Model	Name	User/Org	Task	Library	Dataset	Language	Paper	Related to	License
Facebook/hubert-large-ls960-ft model	hubert-large-ls960-ft	http://facebook	Automatic Speech Recognition	PyTorch TensorFlow Transformers	Librispeech asr Libri-light	English	2106.07447	Audio Speech Hf-asr-leaderboard Hubert Eval Results	Apache-2.0
Jonatasgrosman/wav2vec2-large-xlsr-53-english model	wav2vec2-large-xlsr-53-english	http://jonatasgrosman	Automatic Speech Recognition	PyTorch JAX Safetensors Transformers	Common voice Mozilla-foundation/common voice 6 0	English		Audio Wav2vec2 Speech Hf-asr-leaderboard Mozilla-foundation/common voice 6 0 Robust-speech-event Xlsr-fine-tuning-week Eval Results	Apache-2.0

Tabular Models

See also: Tabular Models

Multimodal Models

See also: Multimodal Models

Model	Name	User/Org	Task	Library	Dataset	Language	Paper	Related to	License
Cambridgeltl/SapBERT-from-PubMedBERT-fulltext model	SapBERT-from-PubMedBERT-fulltext	http://cambridgeltl	Feature Extraction	PyTorch TensorFlow JAX Safetensors Transformers		English	2010.11784	BERT Embedding Biomedical Lexical semantics Bionlp Biology Science Entity linking	Apache-2.0
Facebook/bart-large model	bart-large	http://facebook	Feature Extraction	PyTorch TensorFlow JAX Transformers Rust		English	1910.13461	Bart	Apache-2.0
Facebook/dino-vitb16 model	dino-vitb16	http://facebook	Feature Extraction	PyTorch TensorFlow Transformers	Imagenet-1k		2104.14294	Vision Vit Dino	Apache-2.0
Facebook/dino-vits8 model	dino-vits8	http://facebook	Feature Extraction	PyTorch Transformers	Imagenet-1k		2104.14294	Vision Vit Dino	Apache-2.0
Nlpconnect/vit-gpt2-image-captioning model	vit-gpt2-image-captioning	http://nlpconnect	Image-to-Text	PyTorch Transformers				Vision-encoder-decoder Image-captioning	Apache-2.0
Runwayml/stable-diffusion-v1-5 model	stable-diffusion-v1-5	http://runwayml	Text-to-Image	Diffusers			1910.09700 2103.00020 2207.12598 2112.10752 2205.11487	Stable-diffusion Stable-diffusion-diffusers	Creativeml-openrail-m
Sentence-transformers/all-MiniLM-L6-v2 model	all-MiniLM-L6-v2	http://sentence-transformers	Sentence Similarity Feature Extraction	PyTorch TensorFlow Rust Sentence Transformers	S2orc Flax-sentence-embeddings/stackexchange xml Ms marco Gooaq Yahoo answers topics Code search net Search qa Eli5 Snli Multi nli Wikihow Natural questions Trivia qa Embedding-data/sentence-compression Embedding-data/flickr30k-captions Embedding-data/altlex Embedding-data/simple-wiki Embedding-data/QQP Embedding-data/SPECTER Embedding-data/PAQ pairs Embedding-data/WikiAnswers	English	1904.06472 2102.07033 2104.08727 1704.05179 1810.09305	BERT	Apache-2.0
Sentence-transformers/all-mpnet-base-v2 model	all-mpnet-base-v2	http://sentence-transformers	Sentence Similarity Feature Extraction	PyTorch Sentence Transformers	S2orc Flax-sentence-embeddings/stackexchange xml Gooaq Yahoo answers topics Code search net Search qa Eli5 Snli Multi nli Wikihow Natural questions Trivia qa Embedding-data/sentence-compression Embedding-data/flickr30k-captions Embedding-data/altlex Embedding-data/simple-wiki Embedding-data/QQP Embedding-data/SPECTER Embedding-data/PAQ pairs Embedding-data/WikiAnswers MS Marco	English	1904.06472 2102.07033 2104.08727 1704.05179 1810.09305	Mpnet	Apache-2.0

Reinforcement Learning Models

See also: Reinforcement Learning Models

@@ Line 1: / Line 1: @@
 {{see also|Terms|Applications|Guides}}
+__TOC__
 ==Computer Vision Models==
 {{see also|Computer Vision Models}}