Skip to content

Commit d6fa329

Browse files
vuongminh1907yiyixuxugithub-actions[bot]
authored
update: FluxKontextInpaintPipeline support (#11820)
* update: FluxKontextInpaintPipeline support * fix: Refactor code, remove mask_image_latents and ruff check * feat: Add test case and fix with pytest * Apply style fixes * copies --------- Co-authored-by: YiYi Xu <yixu310@gmail.com> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
1 parent 6f1d669 commit d6fa329

File tree

6 files changed

+1670
-0
lines changed

6 files changed

+1670
-0
lines changed

src/diffusers/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -381,6 +381,7 @@
381381
"FluxFillPipeline",
382382
"FluxImg2ImgPipeline",
383383
"FluxInpaintPipeline",
384+
"FluxKontextInpaintPipeline",
384385
"FluxKontextPipeline",
385386
"FluxPipeline",
386387
"FluxPriorReduxPipeline",
@@ -975,6 +976,7 @@
975976
FluxFillPipeline,
976977
FluxImg2ImgPipeline,
977978
FluxInpaintPipeline,
979+
FluxKontextInpaintPipeline,
978980
FluxKontextPipeline,
979981
FluxPipeline,
980982
FluxPriorReduxPipeline,

src/diffusers/pipelines/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,6 +141,7 @@
141141
"FluxPriorReduxPipeline",
142142
"ReduxImageEncoder",
143143
"FluxKontextPipeline",
144+
"FluxKontextInpaintPipeline",
144145
]
145146
_import_structure["audioldm"] = ["AudioLDMPipeline"]
146147
_import_structure["audioldm2"] = [
@@ -610,6 +611,7 @@
610611
FluxFillPipeline,
611612
FluxImg2ImgPipeline,
612613
FluxInpaintPipeline,
614+
FluxKontextInpaintPipeline,
613615
FluxKontextPipeline,
614616
FluxPipeline,
615617
FluxPriorReduxPipeline,

src/diffusers/pipelines/flux/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,7 @@
3434
_import_structure["pipeline_flux_img2img"] = ["FluxImg2ImgPipeline"]
3535
_import_structure["pipeline_flux_inpaint"] = ["FluxInpaintPipeline"]
3636
_import_structure["pipeline_flux_kontext"] = ["FluxKontextPipeline"]
37+
_import_structure["pipeline_flux_kontext_inpaint"] = ["FluxKontextInpaintPipeline"]
3738
_import_structure["pipeline_flux_prior_redux"] = ["FluxPriorReduxPipeline"]
3839
if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT:
3940
try:
@@ -54,6 +55,7 @@
5455
from .pipeline_flux_img2img import FluxImg2ImgPipeline
5556
from .pipeline_flux_inpaint import FluxInpaintPipeline
5657
from .pipeline_flux_kontext import FluxKontextPipeline
58+
from .pipeline_flux_kontext_inpaint import FluxKontextInpaintPipeline
5759
from .pipeline_flux_prior_redux import FluxPriorReduxPipeline
5860
else:
5961
import sys

0 commit comments

Comments
 (0)