om ModelMixin. Check the superclass documentation for the generic methods the library
implements for all the models (such as downloading or saving, etc.)
For more details, see the original paper: https://arxiv.org/abs/2204.06125

forward

<
source
>
(
hidden_states
timestep: typing.Union[torch.Tensor, float, int]
proj_embedding: FloatTensor
encoder_hidden_states: FloatTensor
attention_mask: typing.Optional[torch.BoolTensor] = None
return_dict: 