ent_channels: int = 4
norm_num_groups: int = 32
sample_size: int = 32
scaling_factor: float = 0.18215
dtype: dtype = <class 'jax.numpy.float32'>
parent: typing.Union[typing.Type[flax.linen.module.Module], typing.Type[flax.core.scope.Scope], typing.Type[flax.linen.module._Sentinel], NoneType] = <flax.linen.module._Sentinel object at 0x7ff2943c0550>
name: str = None

)


Parameters 

in_channels (int, optional, defaults to 3) —
Input channels


o