ype] = <flax.linen.module._Sentinel object at 0x7fa57076a310>
name: str = 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.


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


down_block_types (Tuple[str], optional, defaults to ("CrossAttnDownBlock2D", "CrossAttnDownBlock2D", "CrossAttnD