Skip to content

Commit 9418745

Browse files
committed
SEGA: add semantic guidance for flux pipeline
1 parent 13e8fde commit 9418745

File tree

5 files changed

+1244
-0
lines changed

5 files changed

+1244
-0
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -272,6 +272,7 @@
272272
"FluxImg2ImgPipeline",
273273
"FluxInpaintPipeline",
274274
"FluxPipeline",
275+
"FluxSemanticGuidancePipeline",
275276
"HunyuanDiTControlNetPipeline",
276277
"HunyuanDiTPAGPipeline",
277278
"HunyuanDiTPipeline",
@@ -733,6 +734,7 @@
733734
FluxImg2ImgPipeline,
734735
FluxInpaintPipeline,
735736
FluxPipeline,
737+
FluxSemanticGuidancePipeline,
736738
HunyuanDiTControlNetPipeline,
737739
HunyuanDiTPAGPipeline,
738740
HunyuanDiTPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,7 @@
133133
"FluxImg2ImgPipeline",
134134
"FluxInpaintPipeline",
135135
"FluxPipeline",
136+
"FluxSemanticGuidancePipeline",
136137
]
137138
_import_structure["audioldm"] = ["AudioLDMPipeline"]
138139
_import_structure["audioldm2"] = [
@@ -526,6 +527,7 @@
526527
FluxImg2ImgPipeline,
527528
FluxInpaintPipeline,
528529
FluxPipeline,
530+
FluxSemanticGuidancePipeline,
529531
)
530532
from .hunyuandit import HunyuanDiTPipeline
531533
from .i2vgen_xl import I2VGenXLPipeline

src/diffusers/pipelines/flux/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
_import_structure["pipeline_flux_controlnet_inpainting"] = ["FluxControlNetInpaintPipeline"]
2929
_import_structure["pipeline_flux_img2img"] = ["FluxImg2ImgPipeline"]
3030
_import_structure["pipeline_flux_inpaint"] = ["FluxInpaintPipeline"]
31+
_import_structure["pipeline_semantic_flux"] = ["FluxSemanticGuidancePipeline"]
3132
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
3233
try:
3334
if not (is_transformers_available() and is_torch_available()):
@@ -41,6 +42,7 @@
4142
from .pipeline_flux_controlnet_inpainting import FluxControlNetInpaintPipeline
4243
from .pipeline_flux_img2img import FluxImg2ImgPipeline
4344
from .pipeline_flux_inpaint import FluxInpaintPipeline
45+
from .pipeline_semantic_flux import FluxSemanticGuidancePipeline
4446
else:
4547
import sys
4648

0 commit comments

Comments
 (0)