nal, defaults to  —
obj:("AttnUpBlock2D", "AttnUpBlock2D", "AttnUpBlock2D", "UpBlock2D")): Tuple of upsample block types.


block_out_channels (Tuple[int], optional, defaults to  —
obj:(224, 448, 672, 896)): Tuple of block output channels.


layers_per_block (int, optional, defaults to 2) — The number of layers per block.


mid_block_scale_factor (float, optional, defaults to 1) — The scale factor for the mid block.


downsample_padding (int, o