diff --git a/invokeai/backend/model_management/lora.py b/invokeai/backend/model_management/lora.py index 023772e288..4287072a65 100644 --- a/invokeai/backend/model_management/lora.py +++ b/invokeai/backend/model_management/lora.py @@ -713,6 +713,7 @@ class TextualInversionManager(BaseTextualInversionManager): class ONNXModelPatcher: from .models.base import IAIOnnxRuntimeModel, OnnxRuntimeModel + @classmethod @contextmanager def apply_lora_unet(