u want to cast, and should be False for those you want to skip



Cast the floating-point params to jax.numpy.float16. This returns a new params tree and does not cast the
params in place.
This method can be used on GPU to explicitly convert the model parameters to float16 precision to do full
half-precision training or to save weights in float16 for inference in order to save memory and improve speed.

Examples:


	Copied
>>> from diffusers im