Skip to content

GitLab

Unverified Commit 56d43872 authored by jlonge4's avatar jlonge4 Committed by GitHub
Browse files

feat: add flux kontext (#11985) 

* add flux kontext

* add kontext to img2img

* Apply style fixes
parent edcbe803
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
src/diffusers/pipelines/auto_pipeline.py
View file @ 56d43872
...@@ -49,6 +49,7 @@ from .flux import ( ...@@ -49,6 +49,7 @@ from .flux import (
FluxControlPipeline, FluxControlPipeline,
FluxImg2ImgPipeline, FluxImg2ImgPipeline,
FluxInpaintPipeline, FluxInpaintPipeline,
FluxKontextPipeline,
FluxPipeline, FluxPipeline,
) )
from .hunyuandit import HunyuanDiTPipeline from .hunyuandit import HunyuanDiTPipeline
...@@ -142,6 +143,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -142,6 +143,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("flux", FluxPipeline), ("flux", FluxPipeline),
("flux-control", FluxControlPipeline), ("flux-control", FluxControlPipeline),
("flux-controlnet", FluxControlNetPipeline), ("flux-controlnet", FluxControlNetPipeline),
("flux-kontext", FluxKontextPipeline),
("lumina", LuminaPipeline), ("lumina", LuminaPipeline),
("lumina2", Lumina2Pipeline), ("lumina2", Lumina2Pipeline),
("chroma", ChromaPipeline), ("chroma", ChromaPipeline),
...@@ -171,6 +173,7 @@ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -171,6 +173,7 @@ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING = OrderedDict(
("flux", FluxImg2ImgPipeline), ("flux", FluxImg2ImgPipeline),
("flux-controlnet", FluxControlNetImg2ImgPipeline), ("flux-controlnet", FluxControlNetImg2ImgPipeline),
("flux-control", FluxControlImg2ImgPipeline), ("flux-control", FluxControlImg2ImgPipeline),
("flux-kontext", FluxKontextPipeline),
] ]
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment