d a timestep and returns sample shaped output.
This model inherits from ModelMixin. Check the superclass documentation for the generic methods the library
implements for all the model (such as downloading or saving, etc.)

forward

<
source
>
(
sample: FloatTensor
timestep: typing.Union[torch.Tensor, float, int]
return_dict: bool = True

)
→
UNet2DOutput or tuple

Parameters 

sample (torch.FloatTensor) — (batch, channel, height, width) noisy i