diff --git a/src/diffusers/pipelines/auto_pipeline.py b/src/diffusers/pipelines/auto_pipeline.py index b1a7ffaaea9c..f8b8107cd825 100644 --- a/src/diffusers/pipelines/auto_pipeline.py +++ b/src/diffusers/pipelines/auto_pipeline.py @@ -49,6 +49,7 @@ FluxControlPipeline, FluxImg2ImgPipeline, FluxInpaintPipeline, + FluxKontextPipeline, FluxPipeline, ) from .hunyuandit import HunyuanDiTPipeline @@ -142,6 +143,7 @@ ("flux", FluxPipeline), ("flux-control", FluxControlPipeline), ("flux-controlnet", FluxControlNetPipeline), + ("flux-kontext", FluxKontextPipeline), ("lumina", LuminaPipeline), ("lumina2", Lumina2Pipeline), ("chroma", ChromaPipeline),