ai-models/FLUX.2-klein-base-9b-fp8
2
0
Fork 0
mirror of https://www.modelscope.cn/black-forest-labs/FLUX.2-klein-base-9b-fp8.git synced 2026-04-02 11:02:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update README.md (batch 1/1)

Browse Source
This commit is contained in:
systemd
2026-02-24 12:56:32 +00:00
parent bccaf52504
commit a92b520703
1 changed files with 3 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
README.md
View File

@ -21,59 +21,10 @@ library_name: diffusers
![Teaser](./editing.jpg)
![Teaser](./others.jpg)
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, and runs on consumer hardware.
FLUX.2 [klein] 9B Base is a 9 billion parameter rectified flow transformer capable of generating images from text descriptions and supports multi-reference editing capabilities.
It's a full-capacity foundation model. Undistilled, preserving complete training signal for maximum flexibility. Ideal for fine-tuning, LoRA training, research, and custom pipelines where control matters more than speed. Higher output diversity than the distilled models.
This repository holds an FP8 version of FLUX.2 [klein] Base 9B. The main repository of this model (full BF16 weights) can be found [here](https://huggingface.co/black-forest-labs/FLUX.2-klein-base-9B).
`FLUX.2 [klein] 9B Base` is a 9 billion parameter rectified flow transformer capable of generating images from text descriptions and supports multi-reference editing capabilities.
For more information, please read our [blog post](https://bfl.ai/blog/flux2-klein-towards-interactive-visual-intelligence).
# **Key Features**
1. Undistilled foundation model for maximum flexibility and control.
2. Best suited for creative exploration, fine-tuning, LoRA training, research, and custom image generation or editing pipelines.
3. Trained without step or guidance distillation, making FLUX.2 [klein] 9B Base more efficient and flexible.
4. Open weights for customization and fine-tuning to drive science, research, and empower artists to iterate with speed.
# **Usage**
We provide a reference implementation of FLUX.2 [klein] 9B Base, as well as sampling code, in a dedicated [GitHub repository](https://github.com/black-forest-labs/flux2). Developers and creatives looking to build on top of FLUX.2 [klein] 9B Base are encouraged to use this as a starting point.
FLUX.2 [klein] 9B Base is also available in both [ComfyUI](https://github.com/comfyanonymous/ComfyUI) and [Diffusers](https://github.com/huggingface/diffusers).
## **Using with Diffusers 🧨**
To use FLUX.2 [klein] 9B Base with the 🧨 Diffusers python library, first install or upgrade diffusers:
```shell
pip install -U diffusers
```
Then you can use Flux2KleinPipeline to run the model:
```python
import torch
from diffusers import Flux2KleinPipeline
device = "cuda"
dtype = torch.bfloat16
pipe = Flux2KleinPipeline.from_pretrained("black-forest-labs/FLUX.2-klein-base-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=4.0,
num_inference_steps=50,
generator=torch.Generator(device=device).manual_seed(0)
).images[0]
image.save("flux-klein.png")
```
This repository holds an [FP8 version](https://huggingface.co/black-forest-labs/FLUX.2-klein-base-9b-fp8/blob/main/flux-2-klein-base-9b-fp8.safetensors) of FLUX.2 [klein] 9B Base. The main repository of this model (full BF16 weights) can be found [here](https://huggingface.co/black-forest-labs/FLUX.2-klein-base-9B).
---
Limitations
@ -123,4 +74,4 @@ This model falls under the [FLUX Non-Commercial License](https://huggingface.co/
Trademarks & IP
This project may contain trademarks or logos for projects, products, or services. Use of Black Forest Labs and FLUX trademarks or logos in modified versions of this project must not cause confusion or imply sponsorship or endorsement. Any use of third-party trademarks, intellectual property or logos are subject to those third-party's policies.
This project may contain trademarks or logos for projects, products, or services. Use of Black Forest Labs and FLUX trademarks or logos in modified versions of this project must not cause confusion or imply sponsorship or endorsement. Any use of third-party trademarks, intellectual property or logos are subject to those third-party's policies.