poralModelOutput 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), optional) —
Conditional embeddings for cross attention layer. If not given, cross-attention defaults to
self-attention.

