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 dimensions to use.


sample_size (int, optional) — Pass if the input is discrete. The width of the latent images.
Note that this is fixed at training time as i