 the first element is the sample tensor.



UNet2DConditionOutput
	

class diffusers.models.unet_2d_condition.UNet2DConditionOutput

<
source
>
(
sample: FloatTensor

)


Parameters 

sample (torch.FloatTensor of shape (batch_size, num_channels, height, width)) —
Hidden states conditioned on encoder_hidden_states input. Output of last layer of model.




UNet2DConditionModel
	

class diffusers.UNet2DConditionModel

<
source
>
(
sample_size: typ