Skip to content

feat: add flux2 small decoder support#1402

Merged
leejet merged 3 commits intomasterfrom
flux2_small_decoder
Apr 8, 2026
Merged

feat: add flux2 small decoder support#1402
leejet merged 3 commits intomasterfrom
flux2_small_decoder

Conversation

@leejet
Copy link
Copy Markdown
Owner

@leejet leejet commented Apr 8, 2026

.\bin\Release\sd-cli.exe --diffusion-model  ..\..\ComfyUI\models\diffusion_models\flux-2-klein-4b.safetensors --vae ..\..\ComfyUI\models\vae\full_encoder_small_decoder.safetensors  --llm ..\..\ComfyUI\models\text_encoders\qwen_3_4b.safetensors -p "a lovely cat" --cfg-scale 1.0 --steps 4 -v --offload-to-cpu --diffusion-fa -v
output

@leejet leejet merged commit e8323ca into master Apr 8, 2026
15 checks passed
@stduhpf
Copy link
Copy Markdown
Contributor

stduhpf commented Apr 8, 2026

Here's a quick test comparing output to the original VAE (incuded TAEF2 and an image generated with VAE but saved as jpeg for reference)

- Flux2 VAE JPEG Small Decoder TAEF2
image vaef2 vaef2-j saef2 taef2
diff diff_vaef2 diff_vaef2-j diff_saef2 diff_taef2
MSE 0 4.8037 5.1254 16.0590
PSNR inf dB 41.3151 dB 41.0336 dB 36.0736 dB
SSIM 1 0.9814 0.9791 0.9657
Compute buffer 2562.00 MB 2562.00 MB 2178.00 MB 898.00 MB
decode graph time 2.38s 2.38s 1.73s 0.48s

(using --vae-conv-direct, vulkan backend)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants