aults to None) —
Height and width of input/output sample.


in_channels (int, optional, defaults to 4) — The number of channels in the input sample.


out_channels (int, optional, defaults to 4) — The number of channels in the output.


down_block_types (Tuple[str], optional, defaults to ("CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "DownBlock2D")) —
The tuple of downsample blocks to use.


up_block_types (Tuple[str],