models.vq_model.VQEncoderOutput instead of a plain tuple.   Returns 
VQEncoderOutput or tuple
  
If return_dict is True, a VQEncoderOutput is returned, otherwise a plain tuple
is returned.
  The VQModel forward method.  VQEncoderOutput  class diffusers.models.vq_model.VQEncoderOutput  < source > ( latents: FloatTensor  )   Parameters   latents (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —
The encoded output sample fro