ention_kwargs: typing.Dict[str, typing.Any] = None
attention_mask: typing.Optional[torch.Tensor] = None
encoder_attention_mask: typing.Optional[torch.Tensor] = None
return_dict: bool = True

)
→
Transformer2DModelOutput or tuple

Parameters 

hidden_states ( When discrete, torch.LongTensor of shape (batch size, num latent pixels). —
When continuous, torch.FloatTensor of shape (batch size, channel, height, width)): Input
hidden_states


encoder_