e library
implements for all the models (such as downloading or saving, etc.)

forward

<
source
>
(
sample: FloatTensor
timestep: typing.Union[torch.Tensor, float, int]
encoder_hidden_states: Tensor
class_labels: typing.Optional[torch.Tensor] = None
attention_mask: typing.Optional[torch.Tensor] = None
cross_attention_kwargs: typing.Union[typing.Dict[str, typing.Any], NoneType] = None
return_dict: bool = True

)
→
UNet2DConditionOutput or tuple