to return a ~models.autoencoder_kl.AutoencoderKLOutput instead of a plain tuple.   Returns 
~models.autoencoder_kl.AutoencoderKLOutput or tuple
  
If return_dict is True, a ~models.autoencoder_kl.AutoencoderKLOutput is returned, otherwise a plain
tuple is returned.
  Encode a batch of images using a tiled encoder. When this option is enabled, the VAE will split the input tensor into tiles to compute encoding in several
steps. This is useful to 