ock type


up_block_types (Tuple[str], optional, defaults to (UpDecoderBlock2D)) —
UpDecoder block type


block_out_channels (Tuple[str], optional, defaults to (64,)) —
Tuple containing the number of output channels for each block


layers_per_block (int, optional, defaults to 2) —
Number of Resnet layer for each block


act_fn (str, optional, defaults to silu) —
Activation function


latent_channels (int, optional, defaults to 4) —
Latent spac