Falconsai / image-classification updated 1 year ago

nsfw_image_detection

The Fine-Tuned Vision Transformer (ViT) is a variant of the transformer encoder architecture, similar to BERT, that has been adapted for image classification tasks. This specific model, named "google/vit-base-patch16-224-in21k," is pre-trained on a substantial collection of images in a supervised manner, leveraging the ImageNet-21k dataset. The images in the pre-training dataset are resized to a resolution of 224x224...

Params
90 M
Context
Downloads 30d
5.4 M
Likes
1,154
Commercial use: allowed apache-2.0 Not gated SAFETENSORS View on Hugging Face ↗

Download history

daily snapshots · 10 days
6.6 M5.4 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.

FileQuantSizeEst. VRAMVerdict 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
2023-10-13
Training datasets
undisclosed
Added to our catalog
2026-07-28