Primal VAE released: 80M-param 2x VAE patches to match FLUX 1/2 latents

ostrisai · x · 2026-10-31

Developer ostris open-sourced Primal VAE on Hugging Face in two variants (2x1 and 2x2, compression x channels). It compresses only 2x spatially to 1-2 channels, then DiT-side patch-4 yields latent shapes equivalent to FLUX 1/2 VAEs (16/32 ch @ 8x), beating FLUX 2 VAE on reconstruction. At 80M params, its latent is trained to unit-normal so no scaling/shift factors are needed; images must have sides in multiples of 16. README includes drop-in code for Flux.1/Flux.2 pipelines via diffusers.

Related event: Ostris Open-Sources Primal VAE, Rivals FLUX with Light Compression(2 posts)→

Original post →

More from Multimodal

Multimodal channel →