, os.PathLike]
unet_lora_layers: typing.Dict[str, torch.nn.modules.module.Module] = None
text_encoder_lora_layers: typing.Dict[str, torch.nn.modules.module.Module] = None
is_main_process: bool = True
weight_name: str = None
save_function: typing.Callable = None
safe_serialization: bool = False

)


Parameters 

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


unet_lora_layers (Dict[str, to