distilbert·GPT2LMHeadModel

Distilgpt2 — Hardware Requirements & GPU Compatibility

Chat

DistilGPT-2 is a distilled version of OpenAI's GPT-2 model, compressed to just 88 million parameters while retaining much of the original model's text generation ability. Created using knowledge distillation techniques, it offers significantly faster inference than the full GPT-2 with only a modest reduction in output quality. This model is one of the lightest autoregressive language models available and can run on virtually any hardware, including CPUs. It is a practical choice for educational projects, quick prototyping, and applications where inference speed and minimal resource usage are more important than state-of-the-art generation quality.

2.3M downloads 618 likesFeb 20241K context

Specifications

Publisher
distilbert
Parameters
88M
Architecture
GPT2LMHeadModel
Context Length
1,024 tokens
Vocabulary Size
50,257
Release Date
2024-02-19
License
Apache 2.0

Get Started

How Much VRAM Does Distilgpt2 Need?

Select a quantization to see compatible GPUs below.

QuantizationBitsVRAM
Q2_K3.400.0 GB
Q3_K_S3.500.0 GB
Q3_K_M3.900.1 GB
Q3_K_L4.100.1 GB
IQ4_XS4.300.1 GB
Q4_K_S4.500.1 GB
Q4_K_M4.800.1 GB
Q5_K_S5.500.1 GB
Q5_K_M5.700.1 GB
Q6_K6.600.1 GB
Q8_08.000.1 GB

Which GPUs Can Run Distilgpt2?

Q4_K_M · 0.1 GB

Distilgpt2 (Q4_K_M) requires 0.1 GB of VRAM to load the model weights. For comfortable inference with headroom for KV cache and system overhead, 1+ GB is recommended. 35 GPUs can run it, including NVIDIA GeForce RTX 5090, NVIDIA GeForce RTX 3090 Ti.

Which Devices Can Run Distilgpt2?

Q4_K_M · 0.1 GB

33 devices with unified memory can run Distilgpt2, including NVIDIA DGX H100, NVIDIA DGX A100 640GB.

Related Models

Frequently Asked Questions

How much VRAM does Distilgpt2 need?

Distilgpt2 requires 0.1 GB of VRAM at Q4_K_M, or 0.1 GB at Q8_0.

VRAM = Weights + KV Cache + Overhead

Weights = 88M × 4.8 bits ÷ 8 = 0.1 GB

VRAM usage by quantization

0.1 GB

Learn more about VRAM estimation →

What's the best quantization for Distilgpt2?

For Distilgpt2, Q4_K_M (0.1 GB) offers the best balance of quality and VRAM usage. Q5_K_S (0.1 GB) provides better quality if you have the VRAM. The smallest option is Q2_K at 0.0 GB.

VRAM requirement by quantization

Q2_K
0.0 GB
Q3_K_L
0.1 GB
Q4_K_S
0.1 GB
Q4_K_M
0.1 GB
Q5_K_M
0.1 GB
Q8_0
0.1 GB

★ Recommended — best balance of quality and VRAM usage.

Learn more about quantization →

Can I run Distilgpt2 on a Mac?

Distilgpt2 requires at least 0.0 GB at Q2_K, which exceeds the unified memory of most consumer Macs. You would need a Mac Studio or Mac Pro with a high-memory configuration.

Can I run Distilgpt2 locally?

Yes — Distilgpt2 can run locally on consumer hardware. At Q4_K_M quantization it needs 0.1 GB of VRAM. Popular tools include Ollama, LM Studio, and llama.cpp.

How fast is Distilgpt2?

At Q4_K_M, Distilgpt2 can reach ~48583 tok/s on AMD Instinct MI300X. On NVIDIA GeForce RTX 4090: ~10920 tok/s. Speed depends mainly on GPU memory bandwidth. Real-world results typically within ±20%.

tok/s = (bandwidth GB/s ÷ model GB) × efficiency

Example: AMD Instinct MI300X5300 ÷ 0.1 × 0.55 = ~48583 tok/s

Estimated speed at Q4_K_M (0.1 GB)

~48583 tok/s
~10920 tok/s
~36313 tok/s
~30037 tok/s

Real-world results typically within ±20%. Speed depends on batch size, quantization kernel, and software stack.

Learn more about tok/s estimation →

What's the download size of Distilgpt2?

At Q4_K_M, the download is about 0.05 GB. The full-precision Q8_0 version is 0.09 GB. The smallest option (Q2_K) is 0.04 GB.