safe_serialization (bool, optional, defaults to True) —
Whether to save the model using safetensors or with pickle.    Save attention processor layers to a directory so that it can be reloaded with the
load_attn_procs() method.  Example:   Copied import torch
from diffusers import DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained(
    "CompVis/stable-diffusion-v1-4",
    torch_dtype=torch.float16,
).to("cuda")
pipeline.unet.load_a