ictions and inference.  save_pretrained  < source > ( save_directory: Union params: Union is_main_process: bool = True push_to_hub: bool = False **kwargs  )   Parameters   save_directory (str or os.PathLike) —
Directory to save a model and its configuration file to. Will be created if it doesn’t exist.   params (Union[Dict, FrozenDict]) —
A PyTree of model parameters.   is_main_process (bool, optional, defaults to True) —
Whether the process ca