Model reference · open weights

FLUX.2-klein

Image kp-forks Image edit 1 build Non-commercial 3k dl/mo

FLUX.2-klein is an open-weight image model from kp-forks. FLUX.2-klein-9B (BF16) weighs 34.7 GB; the smallest configuration that runs it is RTX 5090 32 GB.

For self-hosting, compare this build's requirements below with the GPU you plan to rent.

  • FLUX.2-klein-9B is a 9.1 billion parameter image model developed by kp-forks for text-to-image generation and multi-reference editing.
  • It uses a rectified flow transformer architecture with an 8B Qwen3 text embedder and is step-distilled to 4 inference steps for sub-second latency.
  • The model supports English, requires approximately 29GB of VRAM, and is released under a non-commercial license.

Summary of the kp-forks/FLUX.2-klein-9B model card, 2026-10-01

What it is

Released bykp-forks
Released2026-01-22
Parameters9.1B
VRAM34.7 GB for the weights

What it runs on

Memory and cards for FLUX.2-klein-9B (BF16)

34.7 GBweights, file size
18.2 GBbiggest part
5.0 GBworking memory, one 1024² image
537 MBruntime overhead
CardOne imageMemory
RTX 3060 12 GB … RTX 4090 24 GBdoes not fit
RTX 5090 32 GBfits (encoders offloaded)31.0 GB
L40S 48 GBtight44.0 GB
A100 80 GBfits78.2 GB
H100 80 GBfits78.1 GB
RTX PRO 6000 Blackwell 96 GBfits93.8 GB
DGX Spark (GB10) 128 GB unifiedfits107 GB
H200 141 GBfits138 GB
B200 180 GBfits176 GB

From the model card

What kp-forks says about FLUX.2-klein

Read the model card

The FLUX.2 [klein] model family are our fastest image models to date. FLUX.2 [klein] unifies generation and editing in a single compact architecture, delivering state-of-the-art quality with end-to-end inference in as low as under a second. Built for applications that require real-time image generation without sacrificing quality.

FLUX.2 [klein] 9B is a 9 billion parameter rectified flow transformer capable of generating images from text descriptions and supports multi-reference editing capabilities.

Our flagship small model. Defines the Pareto frontier for quality vs. latency across text-to-image, single-reference editing, and multi-reference generation. Matches or exceeds models 5x its size—in under half a second. Built on a 9B flow model with 8B Qwen3 text embedder, step-distilled to 4 inference steps.

For more information, please read our blog post.

Key Features

  1. A distilled model for sub-second image generation with outstanding quality.
  2. Text-to-image and image-to-image multi-reference editing in a single unified model.
  3. Great for real-time generation and integration into applications.
  4. Ideal for creative exploration with excellent prompt adherence and output diversity.
  5. Available for non-commercial use.

Usage

We provide a reference implementation of FLUX.2 [klein] 9B, as well as sampling code, in a dedicated GitHub repository. Developers and creatives looking to build on top of FLUX.2 [klein] 9B are encouraged to use this as a starting point.

API Endpoints

The FLUX.2 [klein] 9B model is available via the BFL API:

FLUX.2 [klein] 9B is also available in both ComfyUI and Diffusers.

Using with Diffusers 🧨

To use FLUX.2 [klein] 9B with the 🧨 Diffusers python library, first install or upgrade diffusers:

pip install -U diffusers

Then you can use Flux2KleinPipeline to run the model:

import torch
from diffusers import Flux2KleinPipeline

device = "cuda"
dtype = torch.bfloat16

pipe = Flux2KleinPipeline.from_pretrained("black-forest-labs/FLUX.2-klein-9B", torch_dtype=dtype)
pipe.enable_model_cpu_offload()  # save some VRAM by offloading the model to CPU

prompt = "A cat holding a sign that says hello world"
image = pipe(
    prompt,
    height=1024,
    width=1024,
    guidance_scale=1.0,
    num_inference_steps=4,
    generator=torch.Generator(device=device).manual_seed(0)
).images[0]
image.save("flux-klein.png")

Limitations

  • This model is not intended or able to provide factual information.
  • While the model can output text, text rendered may be inaccurate or subject to distortion.
  • As a statistical model, this checkpoint may represent or amplify biases observed in the training data.
  • The model may fail to generate output that matches the prompts.
  • Prompt following is heavily influenced by the prompting style.

Out-of-Scope Use

The model and its derivatives may not be used:

  • In any way that violates applicable law.
  • For the purpose of exploiting, harming or attempting to exploit or harm minors in any way; including but not limited to the solicitation, creation, acquisition, or dissemination of child exploitative content.
  • To generate or disseminate deceptive, fraudulent, misleading or otherwise harmful content.
  • To generate or disseminate personal identifiable information that can be used to harm an individual.
  • To harass, abuse, threaten, stalk, or bully individuals or groups of individuals.
  • To create non-consensual intimate imagery or illegal pornographic content.
  • For fully automated decision making or high risk applications that adversely impact an individual's legal rights or otherwise create or modify a binding, enforceable obligation.

Nothing contained in this Model Card should be interpreted as or deemed a restriction or modification to the license the model is released under.

Hardware

The FLUX.2 [klein] 9B model fits in ~29GB VRAM and is accessible on NVIDIA RTX 4090 and above.


Responsible AI Development

Black Forest Labs is committed to the responsible development and deployment of our models. Prior to releasing the FLUX.2 family of models, we evaluated and mitigated a number of risks in our model checkpoints and hosted services, including the generation of unlawful content, including child sexual abuse material (CSAM) and nonconsensual intimate imagery (NCII). We implemented a series of pre-release mitigations to help prevent misuse by third parties, with additional post-release mitigations to help address residual risks:

  1. Pre-training mitigation. We filtered pre-training data for multiple categories of "not safe for work" (NSFW) and known child sexual abuse material (CSAM) to help prevent a user generating unlawful content in response to text prompts or uploaded images. We have partnered with the https://www.iwf.org.uk/, an independent nonprofit organization dedicated to preventing online abuse, to filter known CSAM from the training data.
  2. Post-training mitigation. Subsequently, we undertook multiple rounds of targeted fine-tuning to provide additional mitigation against potential abuse, including both text-to-image (T2I) and image-to-image (I2I) attacks. By inhibiting certain behaviors and suppressing certain concepts in the trained model, these techniques can help to prevent a user generating synthetic CSAM or NCII from a text prompt, or transforming an uploaded image into synthetic CSAM or NCII.
  3. Ongoing evaluation. Throughout this process, we conducted multiple internal and external third-party evaluations of model checkpoints to identify further opportunities for mitigation. External third-party evaluations focused on eliciting CSAM and NCII through adversarial testing with (i) text-only prompts, (ii) a single uploaded refer

Quoted from the model card on Hugging Face. The full card is behind the Hugging Face link above.

Running it yourself

Run it on a rented GPU

Rent a machine by the hour. ComfyUI is installed on it. Open ComfyUI through the tunnel, then Templates → Flux.2 [Klein] 9B: Image Edit, and choose this model's file in its model loader.

# on your rented machine: pip install diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
from diffusers.utils import load_image

pipe = DiffusionPipeline.from_pretrained("kp-forks/FLUX.2-klein-9B", torch_dtype=torch.bfloat16).to("cuda")
start = load_image("/workspace/in.png")
image = pipe(prompt="the same scene at golden hour", image=start).images[0]
image.save("/workspace/out.png")
Renting a GPU: connect, tunnels, Python
# on your rented machine (the ssh line is on its page in the console)
# get REPO FILE FOLDER: one file into /workspace/models/FOLDER, where ComfyUI loads it from
get() { hf download "$1" "$2" --local-dir /workspace/hf-files && mkdir -p "/workspace/models/$3" && mv "/workspace/hf-files/$2" "/workspace/models/$3/$4"; }

# the model (16.9 GB)
get kp-forks/FLUX.2-klein-9B flux-2-klein-9b.safetensors diffusion_models
# what the template also loads “Flux.2 [Klein] 9B: Image Edit” (8.3 GB)
get Comfy-Org/flux2-klein-9B split_files/text_encoders/qwen_3_8b_fp8mixed.safetensors text_encoders
get black-forest-labs/FLUX.2-small-decoder full_encoder_small_decoder.safetensors vae

start-comfyui
Renting a GPU: connect, tunnels, ComfyUI
# on your computer, in a second terminal: ComfyUI in your browser at http://localhost:8188
# HOST, PORT and BOOKING are your machine's, from its page in the console
ssh -o HostKeyAlias=axforge-BOOKING -o StrictHostKeyChecking=accept-new -L 8188:localhost:8188 dev@HOST -p PORT
© 2026 AxForge · EU-hosted AI infrastructure Pricing Docs Trust Privacy Terms