add lucy

Author: Cui-yshoho

mindone/diffusers/__init__.py

@@ -248,6 +248,7 @@
         "LTXImageToVideoPipeline",
         "LTXLatentUpsamplePipeline",
         "LTXPipeline",
+        "LucyEditPipeline",
         "Lumina2Pipeline",
         "Lumina2Text2ImgPipeline",
         "LuminaPipeline",
@@ -640,6 +641,7 @@
         LTXImageToVideoPipeline,
         LTXLatentUpsamplePipeline,
         LTXPipeline,
+        LucyEditPipeline,
         Lumina2Pipeline,
         Lumina2Text2ImgPipeline,
         LuminaPipeline,

mindone/diffusers/pipelines/__init__.py

@@ -148,6 +148,7 @@
     "ltx": ["LTXPipeline", "LTXImageToVideoPipeline", "LTXConditionPipeline", "LTXLatentUpsamplePipeline"],
     "lumina": ["LuminaPipeline", "LuminaText2ImgPipeline"],
     "lumina2": ["Lumina2Pipeline", "Lumina2Text2ImgPipeline"],
+    "lucy": ["LucyEditPipeline"],
     "marigold": [
         "MarigoldDepthPipeline",
         "MarigoldIntrinsicsPipeline",
@@ -393,6 +394,7 @@
         LEditsPPPipelineStableDiffusionXL,
     )
     from .ltx import LTXConditionPipeline, LTXImageToVideoPipeline, LTXLatentUpsamplePipeline, LTXPipeline
+    from .lucy import LucyEditPipeline
     from .lumina import LuminaPipeline, LuminaText2ImgPipeline
     from .lumina2 import Lumina2Pipeline, Lumina2Text2ImgPipeline
     from .marigold import MarigoldDepthPipeline, MarigoldIntrinsicsPipeline, MarigoldNormalsPipeline

mindone/diffusers/pipelines/lucy/__init__.py

@@ -0,0 +1,23 @@
+from typing import TYPE_CHECKING
+
+from ...utils import _LazyModule
+
+_dummy_objects = {}
+_import_structure = {}
+
+_import_structure["pipeline_lucy_edit"] = ["LucyEditPipeline"]
+if TYPE_CHECKING:
+    from .pipeline_lucy_edit import LucyEditPipeline
+
+else:
+    import sys
+
+    sys.modules[__name__] = _LazyModule(
+        __name__,
+        globals()["__file__"],
+        _import_structure,
+        module_spec=__spec__,
+    )
+
+    for name, value in _dummy_objects.items():
+        setattr(sys.modules[__name__], name, value)