Skip to content

Commit 2c574e1

Browse files
authored
add lucy (#1390)
1 parent a690e96 commit 2c574e1

File tree

5 files changed

+754
-0
lines changed

5 files changed

+754
-0
lines changed

mindone/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -252,6 +252,7 @@
252252
"LTXImageToVideoPipeline",
253253
"LTXLatentUpsamplePipeline",
254254
"LTXPipeline",
255+
"LucyEditPipeline",
255256
"Lumina2Pipeline",
256257
"Lumina2Text2ImgPipeline",
257258
"LuminaPipeline",
@@ -648,6 +649,7 @@
648649
LTXImageToVideoPipeline,
649650
LTXLatentUpsamplePipeline,
650651
LTXPipeline,
652+
LucyEditPipeline,
651653
Lumina2Pipeline,
652654
Lumina2Text2ImgPipeline,
653655
LuminaPipeline,

mindone/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,7 @@
150150
"ltx": ["LTXPipeline", "LTXImageToVideoPipeline", "LTXConditionPipeline", "LTXLatentUpsamplePipeline"],
151151
"lumina": ["LuminaPipeline", "LuminaText2ImgPipeline"],
152152
"lumina2": ["Lumina2Pipeline", "Lumina2Text2ImgPipeline"],
153+
"lucy": ["LucyEditPipeline"],
153154
"marigold": [
154155
"MarigoldDepthPipeline",
155156
"MarigoldIntrinsicsPipeline",
@@ -397,6 +398,7 @@
397398
LEditsPPPipelineStableDiffusionXL,
398399
)
399400
from .ltx import LTXConditionPipeline, LTXImageToVideoPipeline, LTXLatentUpsamplePipeline, LTXPipeline
401+
from .lucy import LucyEditPipeline
400402
from .lumina import LuminaPipeline, LuminaText2ImgPipeline
401403
from .lumina2 import Lumina2Pipeline, Lumina2Text2ImgPipeline
402404
from .marigold import MarigoldDepthPipeline, MarigoldIntrinsicsPipeline, MarigoldNormalsPipeline
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
from typing import TYPE_CHECKING
2+
3+
from ...utils import _LazyModule
4+
5+
_dummy_objects = {}
6+
_import_structure = {}
7+
8+
_import_structure["pipeline_lucy_edit"] = ["LucyEditPipeline"]
9+
if TYPE_CHECKING:
10+
from .pipeline_lucy_edit import LucyEditPipeline
11+
12+
else:
13+
import sys
14+
15+
sys.modules[__name__] = _LazyModule(
16+
__name__,
17+
globals()["__file__"],
18+
_import_structure,
19+
module_spec=__spec__,
20+
)
21+
22+
for name, value in _dummy_objects.items():
23+
setattr(sys.modules[__name__], name, value)

0 commit comments

Comments
 (0)