sses is the masked latent pixel. The predicted classes of the unnoised
image do not contain a prediction for the masked pixel as the unnoised image cannot be masked.

forward

<
source
>
(
hidden_states
encoder_hidden_states = None
timestep = None
class_labels = None
cross_attention_kwargs = None
return_dict: bool = True

)
→
Transformer2DModelOutput or tuple

Parameters 

hidden_states ( When discrete, torch.LongTensor of shape (batch size, nu