t for
~models.unet_2d_blocks.CrossAttnDownBlock2D, ~models.unet_2d_blocks.CrossAttnUpBlock2D,
~models.unet_2d_blocks.UNetMidBlock2DCrossAttn.   attention_head_dim (int, optional, defaults to 8) — The dimension of the attention heads.   num_attention_heads (int, optional) —
The number of attention heads. If not defined, defaults to attention_head_dim   resnet_time_scale_shift (str, optional, defaults to "default") — Time scale shift config
for R