update: FluxKontextInpaintPipeline support (#11820) (#1207)

Author: Cui-yshoho

mindone/diffusers/__init__.py

@@ -152,6 +152,7 @@
         "FluxFillPipeline",
         "FluxImg2ImgPipeline",
         "FluxInpaintPipeline",
+        "FluxKontextInpaintPipeline",
         "FluxPipeline",
         "FluxPriorReduxPipeline",
         "HunyuanDiTControlNetPipeline",
@@ -486,6 +487,7 @@
         FluxFillPipeline,
         FluxImg2ImgPipeline,
         FluxInpaintPipeline,
+        FluxKontextInpaintPipeline,
         FluxPipeline,
         FluxPriorReduxPipeline,
         HunyuanDiTControlNetPipeline,

mindone/diffusers/pipelines/__init__.py

@@ -96,6 +96,7 @@
         "FluxPipeline",
         "FluxPriorReduxPipeline",
         "ReduxImageEncoder",
+        "FluxKontextInpaintPipeline",
     ],
     "hunyuandit": ["HunyuanDiTPipeline"],
     "hunyuan_video": [
@@ -326,6 +327,7 @@
         FluxFillPipeline,
         FluxImg2ImgPipeline,
         FluxInpaintPipeline,
+        FluxKontextInpaintPipeline,
         FluxPipeline,
         FluxPriorReduxPipeline,
         ReduxImageEncoder,

mindone/diffusers/pipelines/flux/__init__.py

@@ -17,6 +17,7 @@
     "pipeline_flux_fill": ["FluxFillPipeline"],
     "pipeline_flux_img2img": ["FluxImg2ImgPipeline"],
     "pipeline_flux_inpaint": ["FluxInpaintPipeline"],
+    "pipeline_flux_kontext_inpaint": ["FluxKontextInpaintPipeline"],
     "pipeline_flux_prior_redux": ["FluxPriorReduxPipeline"],
 }
 
@@ -32,6 +33,7 @@
     from .pipeline_flux_fill import FluxFillPipeline
     from .pipeline_flux_img2img import FluxImg2ImgPipeline
     from .pipeline_flux_inpaint import FluxInpaintPipeline
+    from .pipeline_flux_kontext_inpaint import FluxKontextInpaintPipeline
     from .pipeline_flux_prior_redux import FluxPriorReduxPipeline
 else:
     import sys