erimental.  UNet2DConditionOutput  class diffusers.models.unet_2d_condition.UNet2DConditionOutput  < source > ( sample: FloatTensor = None  )   Parameters   sample (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —
The hidden states output conditioned on encoder_hidden_states input. Output of last layer of model.    The output of UNet2DConditionModel.  FlaxUNet2DConditionModel  class diffusers.FlaxUNet2DConditionModel  < s