on  FlaxAutoencoderKLOutput  class diffusers.models.vae_flax.FlaxAutoencoderKLOutput  < source > ( latent_dist: FlaxDiagonalGaussianDistribution  )   Parameters   latent_dist (FlaxDiagonalGaussianDistribution) —
Encoded outputs of Encoder represented as the mean and logvar of FlaxDiagonalGaussianDistribution.
FlaxDiagonalGaussianDistribution allows for sampling latents from the distribution.    Output of AutoencoderKL encoding method.  replace 