ptional, defaults to ("CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "DownBlock2D")) —
The tuple of downsample blocks to use. The corresponding class names will be: “FlaxCrossAttnDownBlock2D”,
“FlaxCrossAttnDownBlock2D”, “FlaxCrossAttnDownBlock2D”, “FlaxDownBlock2D”


block_out_channels (Tuple[int], optional, defaults to (320, 640, 1280, 1280)) —
The tuple of output channels for each block.


layers_per_block (int, opti