 int = 4
down_block_types: typing.Tuple[str] = ('CrossAttnDownBlock2D', 'CrossAttnDownBlock2D', 'CrossAttnDownBlock2D', 'DownBlock2D')
up_block_types: typing.Tuple[str] = ('UpBlock2D', 'CrossAttnUpBlock2D', 'CrossAttnUpBlock2D', 'CrossAttnUpBlock2D')
only_cross_attention: typing.Union[bool, typing.Tuple[bool]] = False
block_out_channels: typing.Tuple[int] = (320, 640, 1280, 1280)
layers_per_block: int = 2
attention_head_dim: typing.Union[int, t