Nunchaku Lite Brings 4-bit Diffusion Inference to Diffusers
Hugging Face
Baidu
Nunchaku Lite, a new integration in Hugging Face Diffusers, allows loading 4-bit quantized diffusion models with a simple from_pretrained() call, without custom pipelines or local CUDA compilation. It reduces VRAM by up to 50% and speeds up inference by about 1.35x, with further gains via torch.compile.
Nunchaku Lite is a new integration path in Hugging Face Diffusers that brings 4-bit inference for diffusion models, based on the SVDQuant quantization method behind the popular Nunchaku inference engine. Unlike standard weight-only quantization, SVDQuant runs main transformer layers with 4-bit weights and activations (W4A4), reducing memory and speeding up the denoising loop. With Nunchaku Lite, loading a quantized checkpoint is as simple as calling from_pretrained() with no local CUDA compilation required, thanks to the Hugging Face kernels package. Two kernel families are used: svdq_w4a4 for attention and MLP projections (available in INT4 and NVFP4 variants) and awq_w4a16 for normalization layers. NVFP4 checkpoints require an NVIDIA Blackwell GPU (RTX 50 series, RTX PRO 6000, B200), while INT4 variants support Turing/Ampere/Ada GPUs (RTX 30 & 40 series, A100, L40S). Benchmarks on an RTX PRO 6000 show a 1.35x speedup and up to 50% VRAM reduction compared to BF16 baseline; combining with torch.compile yields a 1.8x speedup. The companion diffuse-compressor toolkit allows users to quantize their own models and publish them as regular Diffusers repositories.
- Сокращения
- CUDA = Compute Unified Device Architecture
- GPU = Graphics Processing Unit
- VRAM = Video Random Access Memory
- MLP = Multilayer Perceptron
- QKV = Query, Key, Value
- AWQ = Activation-aware Weight Quantization
- NF4 = Normal Float 4
- INT4 = Integer 4-bit
- NVFP4 = NVIDIA 4-bit Floating Point
- BF16 = Brain Floating Point 16
Source: Hugging Face blog —
original
