ntion_dim (int or Tuple[int], optional, defaults to 1280) —
The dimension of the cross attention features.


encoder_hid_dim (int, optional, defaults to None) —
If given, encoder_hidden_states will be projected from this dimension to cross_attention_dim.


attention_head_dim (int, optional, defaults to 8) — The dimension of the attention heads.


resnet_time_scale_shift (str, optional, defaults to "default") — Time scale shift config
for resnet