 —
If specified, weights are saved in the format pytorch_model..bin.



Save a model and its configuration file to a directory, so that it can be re-loaded using the
[from_pretrained()](/docs/diffusers/v0.14.0/en/api/models#diffusers.ModelMixin.from_pretrained) class method.

UNet2DOutput
	

class diffusers.models.unet_2d.UNet2DOutput

<
source
>
(
sample: FloatTensor

)


Parameters 

sample (torch.FloatTensor of shape (batch_size, num_channel