.Tuple[int]] = 8
num_attention_heads: typing.Union[int, typing.Tuple[int], NoneType] = None
cross_attention_dim: int = 1280
dropout: float = 0.0
use_linear_projection: bool = False
dtype: dtype = <class 'jax.numpy.float32'>
flip_sin_to_cos: bool = True
freq_shift: int = 0
controlnet_conditioning_channel_order: str = 'rgb'
conditioning_embedding_out_channels: typing.Tuple[int] = (16, 32, 96, 256)
parent: typing.Union[typing.Type[flax.linen.modul