uple.


Returns

UNet2DOutput or tuple



UNet2DOutput if return_dict is True,
otherwise a tuple. When returning a tuple, the first element is the sample tensor.



UNet1DOutput
	

class diffusers.models.unet_1d.UNet1DOutput

<
source
>
(
sample: FloatTensor

)


Parameters 

sample (torch.FloatTensor of shape (batch_size, num_channels, sample_size)) —
Hidden states output. Output of last layer of model.




UNet1DModel
	

class diffusers.UNet1