 a PIL image.

save_pretrained

<
source
>
(
save_directory: typing.Union[str, os.PathLike]
safe_serialization: bool = False
variant: typing.Optional[str] = None

)


Parameters 

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


safe_serialization (bool, optional, defaults to False) —
Whether to save the model using safetensors or the traditional PyTorch way (that uses pickle).


variant (