Skip to content

Commit 08816ed

Browse files
ighoshsubho蒋硕
authored andcommitted
Feature flux controlnet img2img and inpaint pipeline (huggingface#9408)
* Implemented FLUX controlnet support to Img2Img pipeline
1 parent 5f5bf98 commit 08816ed

File tree

9 files changed

+2634
-1
lines changed

9 files changed

+2634
-1
lines changed

docs/source/en/api/pipelines/flux.md

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,3 +175,16 @@ image.save("flux-fp8-dev.png")
175175
[[autodoc]] FluxInpaintPipeline
176176
- all
177177
- __call__
178+
179+
180+
## FluxControlNetInpaintPipeline
181+
182+
[[autodoc]] FluxControlNetInpaintPipeline
183+
- all
184+
- __call__
185+
186+
## FluxControlNetImg2ImgPipeline
187+
188+
[[autodoc]] FluxControlNetImg2ImgPipeline
189+
- all
190+
- __call__

src/diffusers/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -259,6 +259,8 @@
259259
"CogVideoXPipeline",
260260
"CogVideoXVideoToVideoPipeline",
261261
"CycleDiffusionPipeline",
262+
"FluxControlNetImg2ImgPipeline",
263+
"FluxControlNetInpaintPipeline",
262264
"FluxControlNetPipeline",
263265
"FluxImg2ImgPipeline",
264266
"FluxInpaintPipeline",
@@ -708,6 +710,8 @@
708710
CogVideoXPipeline,
709711
CogVideoXVideoToVideoPipeline,
710712
CycleDiffusionPipeline,
713+
FluxControlNetImg2ImgPipeline,
714+
FluxControlNetInpaintPipeline,
711715
FluxControlNetPipeline,
712716
FluxImg2ImgPipeline,
713717
FluxInpaintPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,8 @@
127127
]
128128
_import_structure["flux"] = [
129129
"FluxControlNetPipeline",
130+
"FluxControlNetImg2ImgPipeline",
131+
"FluxControlNetInpaintPipeline",
130132
"FluxImg2ImgPipeline",
131133
"FluxInpaintPipeline",
132134
"FluxPipeline",
@@ -505,7 +507,14 @@
505507
VersatileDiffusionTextToImagePipeline,
506508
VQDiffusionPipeline,
507509
)
508-
from .flux import FluxControlNetPipeline, FluxImg2ImgPipeline, FluxInpaintPipeline, FluxPipeline
510+
from .flux import (
511+
FluxControlNetImg2ImgPipeline,
512+
FluxControlNetInpaintPipeline,
513+
FluxControlNetPipeline,
514+
FluxImg2ImgPipeline,
515+
FluxInpaintPipeline,
516+
FluxPipeline,
517+
)
509518
from .hunyuandit import HunyuanDiTPipeline
510519
from .i2vgen_xl import I2VGenXLPipeline
511520
from .kandinsky import (

src/diffusers/pipelines/flux/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
else:
2525
_import_structure["pipeline_flux"] = ["FluxPipeline"]
2626
_import_structure["pipeline_flux_controlnet"] = ["FluxControlNetPipeline"]
27+
_import_structure["pipeline_flux_controlnet_image_to_image"] = ["FluxControlNetImg2ImgPipeline"]
28+
_import_structure["pipeline_flux_controlnet_inpainting"] = ["FluxControlNetInpaintPipeline"]
2729
_import_structure["pipeline_flux_img2img"] = ["FluxImg2ImgPipeline"]
2830
_import_structure["pipeline_flux_inpaint"] = ["FluxInpaintPipeline"]
2931
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
@@ -35,6 +37,8 @@
3537
else:
3638
from .pipeline_flux import FluxPipeline
3739
from .pipeline_flux_controlnet import FluxControlNetPipeline
40+
from .pipeline_flux_controlnet_image_to_image import FluxControlNetImg2ImgPipeline
41+
from .pipeline_flux_controlnet_inpainting import FluxControlNetInpaintPipeline
3842
from .pipeline_flux_img2img import FluxImg2ImgPipeline
3943
from .pipeline_flux_inpaint import FluxInpaintPipeline
4044
else:

0 commit comments

Comments
 (0)