UNet2DConditionOutput  
The output of [`UNet2DConditionModel`].  
Args:
sample (`torch.Tensor` of shape `(batch_size, num_channels, height, width)`):
The hidden states output conditioned on `encoder_hidden_states` input. Output of last layer of model.