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 (
FluxControlPipeline,
FluxImg2ImgPipeline,
FluxInpaintPipeline,
FluxKontextPipeline,
FluxPipeline,
)
from .hunyuandit import HunyuanDiTPipeline
......@@ -142,6 +143,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("flux", FluxPipeline),
("flux-control", FluxControlPipeline),
("flux-controlnet", FluxControlNetPipeline),
("flux-kontext", FluxKontextPipeline),
("lumina", LuminaPipeline),
("lumina2", Lumina2Pipeline),
("chroma", ChromaPipeline),
......@@ -171,6 +173,7 @@ AUTO_IMAGE2IMAGE_PIPELINES_MAPPING = OrderedDict(
("flux", FluxImg2ImgPipeline),
("flux-controlnet", FluxControlNetImg2ImgPipeline),
("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