en_states (torch.LongTensor of shape (batch size, num latent pixels) if discrete, torch.FloatTensor of shape (batch size, channel, height, width) if continuous) —
Input hidden_states.   encoder_hidden_states ( torch.LongTensor of shape (batch size, encoder_hidden_states dim), optional) —
Conditional embeddings for cross attention layer. If not given, cross-attention defaults to
self-attention.   timestep ( torch.LongTensor, optional) —
Used to 