


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, torch.nn.Module]) —
State dict of the LoRA layers corresponding to the UNet. Specifying this helps to make the
serialization process easier and cleaner.


text_encoder_lora_layers (Dict[str, torch.nn.Module]) —
State dict of the LoRA layers corresponding to the text_encoder. Since the text_encoder 