Function wrapper

src/diffusers/utils/peft_utils.py:293–320

(self, *args, **kwargs)

    def decorator(forward_fn):
        @functools.wraps(forward_fn)
        def wrapper(self, *args, **kwargs):
            from . import USE_PEFT_BACKEND

            lora_scale = 1.0
            attention_kwargs = kwargs.get(kwargs_name)

            if attention_kwargs is not None:
                attention_kwargs = attention_kwargs.copy()
                kwargs[kwargs_name] = attention_kwargs
                lora_scale = attention_kwargs.pop("scale", 1.0)

                if not USE_PEFT_BACKEND and lora_scale != 1.0:
                    logger.warning(
                        f"Passing `scale` via `{kwargs_name}` when not using the PEFT backend is ineffective."
                    )

            # Apply LoRA scaling if using PEFT backend
            if USE_PEFT_BACKEND:
                scale_lora_layers(self, lora_scale)

            try:
                # Execute the forward pass
                result = forward_fn(self, *args, **kwargs)
                return result
            finally:
                # Always unscale, even if forward pass raises an exception
                if USE_PEFT_BACKEND:
                    unscale_lora_layers(self, lora_scale)

        return wrapper