mentation 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]
class_labels: typing.Optional[torch.Tensor] = None
return_dict: bool = True

)
→
UNet2DOutput or tuple

Parameters 

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


timestep (torch.FloatTensor or float or `in