deprecate lora loader from loaders easily.

2026-01-29 07:22:12 +03:00 · 2025-04-16 15:47:40 +05:30
parent 1b7c286974
commit f0ea9ff2e2
2 changed files with 217 additions and 0 deletions
--- a/src/diffusers/loaders/lora_base.py
+++ b/src/diffusers/loaders/lora_base.py
@@ -0,0 +1,77 @@
+# Copyright 2024 The HuggingFace Team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from ..utils import deprecate
+from .lora.lora_base import LORA_WEIGHT_NAME, LORA_WEIGHT_NAME_SAFE, LoraBaseMixin  # noqa: F401
+
+
+def fuse_text_encoder_lora(text_encoder, lora_scale=1.0, safe_fusing=False, adapter_names=None):
+    from .lora.lora_base import fuse_text_encoder_lora
+
+    deprecation_message = "Importing `fuse_text_encoder_lora()` from diffusers.loaders.lora_base has been deprecated. Please use `from diffusers.loaders.lora.lora_base import fuse_text_encoder_lora` instead."
+    deprecate("diffusers.loaders.lora_base.fuse_text_encoder_lora", "0.36", deprecation_message)
+
+    return fuse_text_encoder_lora(
+        text_encoder, lora_scale=lora_scale, safe_fusing=safe_fusing, adapter_names=adapter_names
+    )
+
+
+def unfuse_text_encoder_lora(text_encoder):
+    from .lora.lora_base import unfuse_text_encoder_lora
+
+    deprecation_message = "Importing `unfuse_text_encoder_lora()` from diffusers.loaders.lora_base has been deprecated. Please use `from diffusers.loaders.lora.lora_base import unfuse_text_encoder_lora` instead."
+    deprecate("diffusers.loaders.lora_base.unfuse_text_encoder_lora", "0.36", deprecation_message)
+
+    return unfuse_text_encoder_lora(text_encoder)
+
+
+def set_adapters_for_text_encoder(
+    adapter_names,
+    text_encoder=None,
+    text_encoder_weights=None,
+):
+    from .lora.lora_base import set_adapters_for_text_encoder
+
+    deprecation_message = "Importing `set_adapters_for_text_encoder()` from diffusers.loaders.lora_base has been deprecated. Please use `from diffusers.loaders.lora.lora_base import set_adapters_for_text_encoder` instead."
+    deprecate("diffusers.loaders.lora_base.set_adapters_for_text_encoder", "0.36", deprecation_message)
+
+    return set_adapters_for_text_encoder(
+        adapter_names=adapter_names, text_encoder=text_encoder, text_encoder_weights=text_encoder_weights
+    )
+
+
+def disable_lora_for_text_encoder(text_encoder=None):
+    from .lora.lora_base import disable_lora_for_text_encoder
+
+    deprecation_message = "Importing `disable_lora_for_text_encoder()` from diffusers.loaders.lora_base has been deprecated. Please use `from diffusers.loaders.lora.lora_base import disable_lora_for_text_encoder` instead."
+    deprecate("diffusers.loaders.lora_base.disable_lora_for_text_encoder", "0.36", deprecation_message)
+
+    return disable_lora_for_text_encoder(text_encoder=text_encoder)
+
+
+def enable_lora_for_text_encoder(text_encoder=None):
+    from .lora.lora_base import enable_lora_for_text_encoder
+
+    deprecation_message = "Importing `enable_lora_for_text_encoder()` from diffusers.loaders.lora_base has been deprecated. Please use `from diffusers.loaders.lora.lora_base import enable_lora_for_text_encoder` instead."
+    deprecate("diffusers.loaders.lora_base.enable_lora_for_text_encoder", "0.36", deprecation_message)
+
+    return enable_lora_for_text_encoder(text_encoder=text_encoder)
+
+
+class LoraBaseMixin(LoraBaseMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `LoraBaseMixin` from diffusers.loaders.lora_base has been deprecated. Please use `from diffusers.loaders.lora.lora_base import LoraBaseMixin` instead."
+        deprecate("diffusers.loaders.lora_base.LoraBaseMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
--- a/src/diffusers/loaders/lora_pipeline.py
+++ b/src/diffusers/loaders/lora_pipeline.py
@@ -0,0 +1,140 @@
+# Copyright 2024 The HuggingFace Team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from ..utils import deprecate
+from .lora.lora_pipeline import (
+    TEXT_ENCODER_NAME,  # noqa: F401
+    TRANSFORMER_NAME,  # noqa: F401
+    UNET_NAME,  # noqa: F401
+    AmusedLoraLoaderMixin,
+    AuraFlowLoraLoaderMixin,
+    CogVideoXLoraLoaderMixin,
+    CogView4LoraLoaderMixin,
+    FluxLoraLoaderMixin,
+    HunyuanVideoLoraLoaderMixin,
+    LTXVideoLoraLoaderMixin,
+    Lumina2LoraLoaderMixin,
+    Mochi1LoraLoaderMixin,
+    SanaLoraLoaderMixin,
+    SD3LoraLoaderMixin,
+    StableDiffusionLoraLoaderMixin,
+    StableDiffusionXLLoraLoaderMixin,
+    WanLoraLoaderMixin,
+)
+
+
+class StableDiffusionLoraLoaderMixin(StableDiffusionLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `StableDiffusionLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import StableDiffusionLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.StableDiffusionLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class StableDiffusionXLLoraLoaderMixin(StableDiffusionXLLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `StableDiffusionXLLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import StableDiffusionXLLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.StableDiffusionXLLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class SD3LoraLoaderMixin(SD3LoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `SD3LoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import SD3LoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.SD3LoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class AuraFlowLoraLoaderMixin(AuraFlowLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `AuraFlowLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import AuraFlowLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.AuraFlowLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class FluxLoraLoaderMixin(FluxLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `FluxLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import FluxLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.FluxLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class AmusedLoraLoaderMixin(AmusedLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `AmusedLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import AmusedLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.AmusedLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class CogVideoXLoraLoaderMixin(CogVideoXLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `CogVideoXLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import CogVideoXLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.CogVideoXLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class Mochi1LoraLoaderMixin(Mochi1LoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `Mochi1LoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import Mochi1LoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.Mochi1LoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class LTXVideoLoraLoaderMixin(LTXVideoLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `LTXVideoLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import LTXVideoLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.LTXVideoLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class SanaLoraLoaderMixin(SanaLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `SanaLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import SanaLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.SanaLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class HunyuanVideoLoraLoaderMixin(HunyuanVideoLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `HunyuanVideoLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import HunyuanVideoLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.HunyuanVideoLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class Lumina2LoraLoaderMixin(Lumina2LoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `Lumina2LoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import Lumina2LoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.Lumina2LoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class WanLoraLoaderMixin(WanLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `WanLoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import WanLoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.WanLoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class CogView4LoraLoaderMixin(CogView4LoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "Importing `CogView4LoraLoaderMixin` from diffusers.loaders.lora_pipeline has been deprecated. Please use `from diffusers.loaders.lora.lora_pipeline import CogView4LoraLoaderMixin` instead."
+        deprecate("diffusers.loaders.lora_pipeline.CogView4LoraLoaderMixin", "0.36", deprecation_message)
+        super().__init__(*args, **kwargs)
+
+
+class LoraLoaderMixin(StableDiffusionLoraLoaderMixin):
+    def __init__(self, *args, **kwargs):
+        deprecation_message = "LoraLoaderMixin is deprecated and this will be removed in a future version. Please use `StableDiffusionLoraLoaderMixin`, instead."
+        deprecate("LoraLoaderMixin", "1.0.0", deprecation_message)
+        super().__init__(*args, **kwargs)