tr], optional, defaults to ("FlaxUpBlock2D", "FlaxCrossAttnUpBlock2D", "FlaxCrossAttnUpBlock2D", "FlaxCrossAttnUpBlock2D")) —
The tuple of upsample blocks to use.   mid_block_type (str, optional, defaults to "UNetMidBlock2DCrossAttn") —
Block type for middle of UNet, it can be one of UNetMidBlock2DCrossAttn. If None, the mid block layer is skipped.   block_out_channels (Tuple[int], optional, defaults to (320, 640, 1280, 1280)) —
The tuple of ou