google / image-classification updated 2 years ago

vit-base-patch16-224

Name: vit-base-patch16-224
Author: google

Vision Transformer (ViT) model pre-trained on ImageNet-21k (14 million images, 21,843 classes) at resolution 224x224, and fine-tuned on ImageNet 2012 (1 million images, 1,000 classes) at resolution 224x224. It was introduced in the paper An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale by Dosovitskiy et al. and first released in this repository. However, the weights were converted from the t...

Params

90 M

Context

—

Downloads 30d

4.7 M

Likes

990

Commercial use: allowed apache-2.0 Not gated SAFETENSORS View on Hugging Face ↗

Download history

daily snapshots · 10 days

4.7 M4.5 M

Jul 28Jul 31Aug 3Aug 6

Can you run it?

Estimated VRAM at 8K context unless noted. Pick your hardware to see the verdict per quantization.

File	Quant	Size	Est. VRAM	Verdict on RTX 4090 · 24 GB
model.safetensors	f32	0.3 GB	0.9 GB	✅ Runs comfortably

Estimate: file size × 1.1 + KV cache at 8K + 0.5 GB overhead. Not a benchmark — how we calculate this.

Specifications

Architecture: ViTForImageClassification
Parameters: 90 M
Tensor type: F32
Layers / heads: 12 / 12
Licence: apache-2.0
First seen on the Hub: 2022-03-02
Training datasets: imagenet-1k, imagenet-21k
Added to our catalog: 2026-07-28

Compare with

mobilenetv3_small_100.lamb_in1k 18.5 M nsfw_image_detection 5.4 M fairface_age_image_detection 3.2 M resnet50.a1_in1k 2.8 M mit-b2 1.6 M gender-classification 1.5 M

Appears in

Best open coding models Commercial-use text models Models that fit in 12 GB