distribution.    Output of AutoencoderKL encoding method.  DecoderOutput  class diffusers.models.autoencoders.vae.DecoderOutput  < source > ( sample: FloatTensor  )   Parameters   sample (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —
The decoded output sample from the last layer of the model.    Output of decoding method.  FlaxAutoencoderKL  class diffusers.FlaxAutoencoderKL  < source > ( in_channels: int = 3 out_chann