eturns 
~models.unet_2d.UNet2DOutput or tuple
  
If return_dict is True, an ~models.unet_2d.UNet2DOutput is returned, otherwise a tuple is
returned where the first element is the sample tensor.
  The UNet2DModel forward method.  UNet2DOutput  class diffusers.models.unets.unet_2d.UNet2DOutput  < source > ( sample: FloatTensor  )   Parameters   sample (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —
The hidden states outpu