ucture as the params tree. The leaves should be booleans, True for params
you want to cast, and should be False for those you want to skip



Cast the floating-point params to jax.numpy.float32. This method can be used to explicitly convert the
model parameters to fp32 precision. This returns a new params tree and does not cast the params in place.

Examples:


	Copied
>>> from diffusers import FlaxUNet2DConditionModel

>>> # Download model and