PulseAugur
EN
LIVE 18:47:55

FLUX.2-9B-klein text encoder pruned by 38% to fit 16GB VRAM

A user has successfully reduced the size of the FLUX.2-9B-klein model's text encoder by 38%, resulting in a 5.1B parameter version. This optimization allows the entire fp8 model to fit within 16GB of VRAM for 768x768 image generation without requiring text encoder offloading. The pruned model maintains a high embedding fidelity of 0.975, with detailed technical steps including layer merging, FFN width reduction, and KV group optimization. AI

IMPACT Enables lower-resource hardware to run advanced image generation models.

RANK_REASON User-driven optimization of an existing open-source model's components. [lever_c_demoted from research: ic=1 ai=1.0]

Read on r/StableDiffusion →

AI-generated summary · Google Gemini · from 1 sources. How we write summaries →

FLUX.2-9B-klein text encoder pruned by 38% to fit 16GB VRAM

COVERAGE [1]

  1. r/StableDiffusion TIER_2 English(EN) · /u/ThaJedi ·

    I removed 38% of FLUX.2-9B-klein's text encoder. For 768x768 image generation the whole fp8 model fits in 16GB VRAM without text encoder offloading. Weights + ComfyUI nodes

    <!-- SC_OFF --><div class="md"><p><strong>What you get:</strong> a drop in replacement for the 8.2B Qwen3 text encoder at <strong>5.10B parameters (38% smaller)</strong>. DiT and VAE untouched. For 768x768 generation the whole fp8 pipeline now fits in 16 GB VRAM without text enco…