lock2D', 'UpBlock2D')
block_out_channels: typing.Tuple[int] = (224, 448, 672, 896)
layers_per_block: int = 2
mid_block_scale_factor: float = 1
downsample_padding: int = 1
act_fn: str = 'silu'
attention_head_dim: typing.Optional[int] = 8
norm_num_groups: int = 32
norm_eps: float = 1e-05
resnet_time_scale_shift: str = 'default'
add_attention: bool = True
class_embed_type: typing.Optional[str] = None
num_class_embeds: typing.Optional[int] = None

