Function disable_lora_for_text_encoder

src/diffusers/loaders/lora_base.py:163–174

Disables the LoRA layers for the text encoder. Args: text_encoder (`torch.nn.Module`, *optional*): The text encoder module to disable the LoRA layers for. If `None`, it will try to get the `text_encoder` attribute.

(text_encoder: "PreTrainedModel" | None = None)

def disable_lora_for_text_encoder(text_encoder: "PreTrainedModel" | None = None):
    """
    Disables the LoRA layers for the text encoder.

    Args:
        text_encoder (`torch.nn.Module`, *optional*):
            The text encoder module to disable the LoRA layers for. If `None`, it will try to get the `text_encoder`
            attribute.
    """
    if text_encoder is None:
        raise ValueError("Text Encoder not found.")
    set_adapter_layers(text_encoder, enabled=False)


def enable_lora_for_text_encoder(text_encoder: "PreTrainedModel" | None = None):