(for example purposes, not runnable).
>>> model, params = FlaxUNet2DConditionModel.from_pretrained("./test/saved_model/")

save_pretrained

<
source
>
(
save_directory: typing.Union[str, os.PathLike]
params: typing.Union[typing.Dict, flax.core.frozen_dict.FrozenDict]
is_main_process: bool = True

)


Parameters 

save_directory (str or os.PathLike) —
Directory to which to save. Will be created if it doesn’t exist.


params (Union[Dict, FrozenDi