ffusers/v0.16.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 (Union[Dict, FrozenDict]) —
A PyTree of model parameters.


mask (Union[Dict, FrozenDict]) —
A PyTree with same structure as the params tree. The leaves should be booleans, True for params
you want to cast, and should be Fa