Hugging Face's logo

AEmotionStudio
/
flux-klein

Image-to-Image
Diffusers
Safetensors
Flux2KleinPipeline
flux
inpainting
image-editing
diffusion
Model card Files
xet
Community
flux-klein
File size: 2,980 Bytes
bf0a5c0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
 
---
license: apache-2.0
pipeline_tag: image-to-image
tags:
  - flux
  - inpainting
  - image-editing
  - diffusion
  - safetensors
base_model: black-forest-labs/FLUX.2-klein-4B
---

# FLUX Klein 4B — AEmotionStudio Mirror

**Mirror of [black-forest-labs/FLUX.2-klein-4B](https://huggingface.co/black-forest-labs/FLUX.2-klein-4B)** for use with [ComfyUI-FFMPEGA](https://github.com/AEmotionStudio/ComfyUI-FFMPEGA).

## About

FLUX Klein is a 4-billion parameter image editing and generation model from [Black Forest Labs](https://blackforestlabs.ai/). It supports:

- **Object removal** — erase unwanted objects from images/video frames
- **Text-guided editing** — change hair color, swap backgrounds, modify clothing, etc.
- **Inpainting** — fill masked regions with contextually appropriate content

This mirror provides the same bf16 safetensors weights as the upstream repo, hosted on AEmotionStudio infrastructure for supply-chain resilience.

## Model Details

| Property | Value |
|----------|-------|
| **Architecture** | FLUX (Rectified Flow Transformer) |
| **Parameters** | 4B |
| **Precision** | bf16 (bfloat16) |
| **Total Size** | ~15 GB |
| **License** | [Apache 2.0](https://www.apache.org/licenses/LICENSE-2.0) |
| **Source** | [black-forest-labs/FLUX.2-klein-4B](https://huggingface.co/black-forest-labs/FLUX.2-klein-4B) |

### Files

| Component | File | Size |
|-----------|------|------|
| Transformer | `transformer/diffusion_pytorch_model.safetensors` | 7.4 GB |
| Text Encoder (T5-XXL) | `text_encoder/model-00001-of-00002.safetensors` | 4.7 GB |
| Text Encoder (T5-XXL) | `text_encoder/model-00002-of-00002.safetensors` | 2.9 GB |
| VAE | `vae/diffusion_pytorch_model.safetensors` | 160 MB |
| Tokenizer, configs | `tokenizer/`, `scheduler/`, `*.json` | < 15 MB |

## Usage with ComfyUI-FFMPEGA

This model is **auto-downloaded** when you use the `auto_mask` skill with `effect=remove` or `effect=edit` in [ComfyUI-FFMPEGA](https://github.com/AEmotionStudio/ComfyUI-FFMPEGA).

### Object Removal

Prompt the FFMPEGA Agent:
```
Remove the watermark
```

Or use the Effects Builder with SAM3 target + `remove` effect.

### Text-Guided Editing

Prompt the FFMPEGA Agent:
```
Change the hair to red
```

Or use `auto_mask` with `effect=edit` and `edit_prompt="change the hair to red"`.

### Manual Download

If auto-download is disabled, download all files and place them in:
```
ComfyUI/models/flux_klein/
```

## VRAM Requirements

- **Minimum**: ~8 GB (with aggressive CPU offloading)
- **Recommended**: ~13 GB (bf16/fp16, sequential CPU offload)
- GPUs without native bf16 support (pre-Ampere) automatically use fp16 at load time

## License

The model weights are licensed under **Apache 2.0** by Black Forest Labs. See [LICENSE.md](LICENSE.md) for full terms.

## Citation

```bibtex
@misc{flux2klein2025,
  title={FLUX.2 Klein},
  author={Black Forest Labs},
  year={2025},
  url={https://huggingface.co/black-forest-labs/FLUX.2-klein-4B}
}
```