stephenbtl commited on
Commit
a3efc24
·
verified ·
1 Parent(s): fee117d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +7 -4
README.md CHANGED
@@ -12,12 +12,15 @@ tags:
12
 
13
  ![Comparison Panel](./comparison_panel.jpeg)
14
 
15
- This repository holds a small (distilled) VAE decoder for FLUX.2. It is a drop-in replacement for the full decoder, with narrower channel widths (`[96, 192, 384, 384]` vs `[128, 256, 512, 512]`) for faster decoding with minimal to zero quality loss. The encoder is unchanged.
16
 
17
- - **~28M** decoder params (vs ~50M full decoder)
18
- - **~1.4x** faster decode
19
- - **~1.4x** less VRAM
20
 
 
 
 
 
 
21
 
22
  Compatible with all open FLUX.2 models:
23
  - [FLUX.2-klein-4B](https://huggingface.co/black-forest-labs/FLUX.2-klein-4B)
 
12
 
13
  ![Comparison Panel](./comparison_panel.jpeg)
14
 
15
+ `FLUX.2 Small Decoder` is a distilled VAE decoder that serves as a **drop-in replacement** for the standard FLUX.2 decoder. It delivers faster decoding and lower VRAM usage with minimal to zero quality loss. The encoder remains unchanged.
16
 
17
+ # **Key Features**
 
 
18
 
19
+ 1. **~1.4x faster decoding** compared to the full decoder.
20
+ 2. **~1.4x less VRAM** at decode time, enabling higher resolutions without running out of memory.
21
+ 3. **~28M decoder parameters** (vs ~50M in the full decoder) thanks to narrower channel widths (`[96, 192, 384, 384]` vs `[128, 256, 512, 512]`).
22
+ 4. **Minimal quality loss** — images are almost identical.
23
+ 5. Available under the Apache 2.0 license.
24
 
25
  Compatible with all open FLUX.2 models:
26
  - [FLUX.2-klein-4B](https://huggingface.co/black-forest-labs/FLUX.2-klein-4B)