flux2_vae_decoder

Packages

chatterbox
- chatterbox Reference
cornfab
- cornfab Reference
diffuseR
earshot
- run_app
fyi
- fyi Reference
llamaR
- llamaR Reference
rapt
- rapt Reference
rformat
- rformat Reference
saber
- saber Reference
stt.api
tinypkgr
- tinypkgr Reference
tinyrox
- tinyrox Reference
tts.api
- tts.api Reference
whisper

FLUX.2 VAE decoder

Description

The AutoencoderKLFlux2 decode path: post_quant_conv (1x1, 32 channels) followed by the standard AutoencoderKL decoder body (reused from vae_decoder_native), plus the BatchNorm running statistics used for latent (de)normalization. Reference: src/diffusers/models/autoencoders/autoencoder_kl_flux2.py.

Usage

flux2_vae_decoder(latent_channels = 32L,
                  block_channels = c(512L, 512L, 256L, 128L), norm_groups = 32L)

Arguments

latent_channels: Integer (32 for FLUX.2).
block_channels: Decoder block channels (reversed encoder block_out_channels).
norm_groups: Integer. Group norm groups.

Value

Module whose forward(z) decodes [B, 32, H, W] latents to [B, 3, 8H, 8W] images; $bn$running_mean / $bn$running_var carry the normalization statistics.