s: int = 64 projection_class_embeddings_input_dim: Optional = None 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.   out_channels (int, optional, defaults to 4) —
The number of channels in the output.   down_block_types (Tuple[str], o