Skip to content

GitLab

Unverified Commit 7f51f286 authored by Beinsezii's avatar Beinsezii Committed by GitHub
Browse files

Add Hunyuan AutoPipe mapping (#8505)

parent 829f6def
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
src/diffusers/pipelines/auto_pipeline.py
View file @ 7f51f286
...@@ -27,6 +27,7 @@ from .controlnet import ( ...@@ -27,6 +27,7 @@ from .controlnet import (
StableDiffusionXLControlNetPipeline, StableDiffusionXLControlNetPipeline,
) )
from .deepfloyd_if import IFImg2ImgPipeline, IFInpaintingPipeline, IFPipeline from .deepfloyd_if import IFImg2ImgPipeline, IFInpaintingPipeline, IFPipeline
from .hunyuandit import HunyuanDiTPipeline
from .kandinsky import ( from .kandinsky import (
KandinskyCombinedPipeline, KandinskyCombinedPipeline,
KandinskyImg2ImgCombinedPipeline, KandinskyImg2ImgCombinedPipeline,
...@@ -70,6 +71,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict( ...@@ -70,6 +71,7 @@ AUTO_TEXT2IMAGE_PIPELINES_MAPPING = OrderedDict(
("stable-diffusion-xl", StableDiffusionXLPipeline), ("stable-diffusion-xl", StableDiffusionXLPipeline),
("stable-diffusion-3", StableDiffusion3Pipeline), ("stable-diffusion-3", StableDiffusion3Pipeline),
("if", IFPipeline), ("if", IFPipeline),
("hunyuan", HunyuanDiTPipeline),
("kandinsky", KandinskyCombinedPipeline), ("kandinsky", KandinskyCombinedPipeline),
("kandinsky22", KandinskyV22CombinedPipeline), ("kandinsky22", KandinskyV22CombinedPipeline),
("kandinsky3", Kandinsky3Pipeline), ("kandinsky3", Kandinsky3Pipeline),
......
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