he input is continuous).   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) — The width of the latent images (specify if the input is discrete).
This is fixed during training since it is used to lea