ls: Tuple = (64,) layers_per_block: int = 1 act_fn: str = 'silu' latent_channels: int = 4 norm_num_groups: int = 32 sample_size: int = 32 scaling_factor: float = 0.18215 dtype: dtype = <class 'jax.numpy.float32'> parent: Union = <flax.linen.module._Sentinel object at 0x7fbd557629e0> name: Optional = None  )   Parameters   in_channels (int, optional, defaults to 3) —
Number of channels in the input image.   out_channels (int, optional, defaults 