_dim (int, optional, defaults to 88) — The number of channels in each head.


in_channels (int, optional) —
Pass if the input is continuous. The number of channels in the input and output.


num_layers (int, optional, defaults to 1) — The number of layers of Transformer blocks to use.


dropout (float, optional, defaults to 0.0) — The dropout probability to use.


cross_attention_dim (int, optional) — The number of encoder_hidden_states dimensi