Skip to main content

vae_decode_chain

cortiq_engine::gpu

Function vae_decode_chain 

Source 
pub fn vae_decode_chain(
    a: &VaeChainArgs<'_>,
    z: &[f32],
    h: usize,
    w: usize,
    out: &mut [f32],
) -> bool
Expand description

Resident Flux-VAE decode (the whole decoder on the device, one latent upload, one RGB readback). a comes from VaeDecoder::chain_args(). z is [latent_channels, h, w] ALREADY de-normalised (z/scaling_factor + shift_factor — the conv_in input); out is [3, 8h, 8w], the raw decoder output (≈[-1, 1], before x/2+0.5).