if each TransformerBlock should contain two self-attention layers



Transformer model for video-like data.

forward

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

)
→
~models.transformer_2d.TransformerTemporalModelOutput or tuple

Parameters 

hidden_states ( When discrete, torch.LongTensor of shape (batch size, num latent pixel