ApX logoApX logo

FLUX.2 Klein 4B

Parameters

4B

Modality

Image

Auxiliary Parameters

80M

Architecture

Diffusion Transformer

License

Apache 2.0

Release Date

10 Feb 2025

API Pricing (per 1M)

Self-hosted only

Hardware Sizing

Precision

Workload Memory: 4.66 GB

INT4 · 1024x1024

DiT Backbone Weights: 1.86 GB

VAE Decode Workspace: 0.29 GB

Runtime & Graphs: 2.50 GB

Minimum Required Hardware

1x NVIDIA L4 24GB

4.7 / 21.6 GB usable

1x RTX 3060 12GB

4.7 / 10.8 GB usable

Apple M4 24GB

4.7 / 18.0 GB usable

Determine requirements for your specific workload:

Open in VRAM Calculator

Architecture Diagram

Noisy Image LatentsLatent Image Space (VAE)Patch Embedding2D Latent PatchifyHidden: 7.7k · Text: 80M enc · Timestep (t)x 25 DiT blocksAdaLN ModulationNorm · Timestep scale/shiftJoint / Cross-Attention24 headsHead dim: 128+AdaLN ModulationNorm · Timestep scale/shiftFeed-Forward (MLP)Activation+Final Norm (AdaLN Scale)Predicted Noise / LatentsDenoised Latents to VAE

About FLUX.2 Klein 4B

FLUX.2 Klein 4B is a compact 4B parameter diffusion transformer by Black Forest Labs, engineered for high-throughput edge deployment and consumer GPUs with minimal memory consumption.

Technical Specifications

Attention

Attention Heads

24

Attention Head Dimension

128

Dimensions

Auxiliary Parameters

80M

Hidden Dimension Size

7,680

Number of Layers

25

About Flux 2


Other Flux 2 Models