From 0cb92717f9f46731cd0df8f48bf0664a3e0e34cf Mon Sep 17 00:00:00 2001 From: CyrusVorwald <90732384+CyrusVorwald@users.noreply.github.com> Date: Tue, 2 Jan 2024 08:44:48 -0500 Subject: [PATCH] add StableDiffusionXLControlNetInpaintPipeline to auto pipeline (#6302) * add StableDiffusionXLControlNetInpaintPipeline to auto pipeline * fixed style --- src/diffusers/pipelines/auto_pipeline.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/pipelines/auto_pipeline.py b/src/diffusers/pipelines/auto_pipeline.py index 00738be3f3..a87404c693 100644 --- a/src/diffusers/pipelines/auto_pipeline.py +++ b/src/diffusers/pipelines/auto_pipeline.py @@ -24,6 +24,7 @@ from .controlnet import ( StableDiffusionControlNetInpaintPipeline, StableDiffusionControlNetPipeline, StableDiffusionXLControlNetImg2ImgPipeline, + StableDiffusionXLControlNetInpaintPipeline, StableDiffusionXLControlNetPipeline, ) from .deepfloyd_if import IFImg2ImgPipeline, IFInpaintingPipeline, IFPipeline @@ -97,6 +98,7 @@ AUTO_INPAINT_PIPELINES_MAPPING = OrderedDict( ("kandinsky", KandinskyInpaintCombinedPipeline), ("kandinsky22", KandinskyV22InpaintCombinedPipeline), ("stable-diffusion-controlnet", StableDiffusionControlNetInpaintPipeline), + ("stable-diffusion-xl-controlnet", StableDiffusionXLControlNetInpaintPipeline), ] )