Skip to content

Commit 45286fd

Browse files
committed
add flux fill pipeline
1 parent 513c334 commit 45286fd

File tree

4 files changed

+918
-0
lines changed

4 files changed

+918
-0
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -275,6 +275,7 @@
275275
"FluxImg2ImgPipeline",
276276
"FluxInpaintPipeline",
277277
"FluxPipeline",
278+
"FluxFillPipeline",
278279
"HunyuanDiTControlNetPipeline",
279280
"HunyuanDiTPAGPipeline",
280281
"HunyuanDiTPipeline",
@@ -740,6 +741,7 @@
740741
FluxImg2ImgPipeline,
741742
FluxInpaintPipeline,
742743
FluxPipeline,
744+
FluxFillPipeline,
743745
HunyuanDiTControlNetPipeline,
744746
HunyuanDiTPAGPipeline,
745747
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+
"FluxFillPipeline",
136137
]
137138
_import_structure["audioldm"] = ["AudioLDMPipeline"]
138139
_import_structure["audioldm2"] = [
@@ -527,6 +528,7 @@
527528
FluxImg2ImgPipeline,
528529
FluxInpaintPipeline,
529530
FluxPipeline,
531+
FluxFillPipeline,
530532
)
531533
from .hunyuandit import HunyuanDiTPipeline
532534
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_flux_fill"] = ["FluxFillPipeline"]
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_flux_fill import FluxFillPipeline
4446
else:
4547
import sys
4648

0 commit comments

Comments
 (0)