parent: Union = <flax.linen.module._Sentinel object at 0x7fbece3b2b00> name: Optional = None  )   Parameters   sample_size (int, optional) —
The size of the input sample.   in_channels (int, optional, defaults to 4) —
The number of channels in the input sample.   down_block_types (Tuple[str], optional, defaults to ("FlaxCrossAttnDownBlock2D", "FlaxCrossAttnDownBlock2D", "FlaxCrossAttnDownBlock2D", "FlaxDownBlock2D")) —
The tuple of downsample b