tates = 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, num latent pixels). —
When continous, torch.FloatTensor of shape (batch size, channel, height, width)): Input
hidden_states


encoder_hidden_states ( torch.LongTensor of shape (batch size, encoder_hidden_states dim), opti