esses. In this case, set is_main_process=True only on
the main process to avoid race conditions.



Save a model and its configuration file to a directory, so that it can be re-loaded using the
[from_pretrained()](/docs/diffusers/v0.14.0/en/api/models#diffusers.FlaxModelMixin.from_pretrained) class method

to_bf16

<
source
>
(
params: typing.Union[typing.Dict, flax.core.frozen_dict.FrozenDict]
mask: typing.Any = None

)


Parameters 

params (