Skip to content

Commit 249a9e4

Browse files
Gothosyiyixuxu
andauthored
Add Flux inpainting and Flux Img2Img (#9135)
--------- Co-authored-by: yiyixuxu <[email protected]>
1 parent 2ee3215 commit 249a9e4

File tree

9 files changed

+2210
-2
lines changed

9 files changed

+2210
-2
lines changed

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

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -163,3 +163,15 @@ image.save("flux-fp8-dev.png")
163163
[[autodoc]] FluxPipeline
164164
- all
165165
- __call__
166+
167+
## FluxImg2ImgPipeline
168+
169+
[[autodoc]] FluxImg2ImgPipeline
170+
- all
171+
- __call__
172+
173+
## FluxInpaintPipeline
174+
175+
[[autodoc]] FluxInpaintPipeline
176+
- all
177+
- __call__

src/diffusers/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -258,6 +258,8 @@
258258
"CogVideoXVideoToVideoPipeline",
259259
"CycleDiffusionPipeline",
260260
"FluxControlNetPipeline",
261+
"FluxImg2ImgPipeline",
262+
"FluxInpaintPipeline",
261263
"FluxPipeline",
262264
"HunyuanDiTControlNetPipeline",
263265
"HunyuanDiTPAGPipeline",
@@ -703,6 +705,8 @@
703705
CogVideoXVideoToVideoPipeline,
704706
CycleDiffusionPipeline,
705707
FluxControlNetPipeline,
708+
FluxImg2ImgPipeline,
709+
FluxInpaintPipeline,
706710
FluxPipeline,
707711
HunyuanDiTControlNetPipeline,
708712
HunyuanDiTPAGPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,12 @@
124124
"AnimateDiffSparseControlNetPipeline",
125125
"AnimateDiffVideoToVideoPipeline",
126126
]
127-
_import_structure["flux"] = ["FluxPipeline", "FluxControlNetPipeline"]
127+
_import_structure["flux"] = [
128+
"FluxControlNetPipeline",
129+
"FluxImg2ImgPipeline",
130+
"FluxInpaintPipeline",
131+
"FluxPipeline",
132+
]
128133
_import_structure["audioldm"] = ["AudioLDMPipeline"]
129134
_import_structure["audioldm2"] = [
130135
"AudioLDM2Pipeline",
@@ -494,7 +499,7 @@
494499
VersatileDiffusionTextToImagePipeline,
495500
VQDiffusionPipeline,
496501
)
497-
from .flux import FluxControlNetPipeline, FluxPipeline
502+
from .flux import FluxControlNetPipeline, FluxImg2ImgPipeline, FluxInpaintPipeline, FluxPipeline
498503
from .hunyuandit import HunyuanDiTPipeline
499504
from .i2vgen_xl import I2VGenXLPipeline
500505
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_img2img"] = ["FluxImg2ImgPipeline"]
28+
_import_structure["pipeline_flux_inpaint"] = ["FluxInpaintPipeline"]
2729
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
2830
try:
2931
if not (is_transformers_available() and is_torch_available()):
@@ -33,6 +35,8 @@
3335
else:
3436
from .pipeline_flux import FluxPipeline
3537
from .pipeline_flux_controlnet import FluxControlNetPipeline
38+
from .pipeline_flux_img2img import FluxImg2ImgPipeline
39+
from .pipeline_flux_inpaint import FluxInpaintPipeline
3640
else:
3741
import sys
3842

0 commit comments

Comments
 (0)