From 9640e7667a705882b0e74fae0c46a373e69d48a8 Mon Sep 17 00:00:00 2001 From: Alan Bedian Date: Tue, 9 Jul 2024 19:19:39 -0700 Subject: [PATCH] Add StableDiffusion3InpaintPipeline to AutoPipelineForInpainting --- 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 3eb98cfef912..fbb90f94d531 100644 --- a/src/diffusers/pipelines/auto_pipeline.py +++ b/src/diffusers/pipelines/auto_pipeline.py @@ -62,6 +62,7 @@ ) from .stable_diffusion_3 import ( StableDiffusion3Img2ImgPipeline, + StableDiffusion3InpaintPipeline, StableDiffusion3Pipeline, ) from .stable_diffusion_xl import ( @@ -115,6 +116,7 @@ [ ("stable-diffusion", StableDiffusionInpaintPipeline), ("stable-diffusion-xl", StableDiffusionXLInpaintPipeline), + ("stable-diffusion-3", StableDiffusion3InpaintPipeline), ("if", IFInpaintingPipeline), ("kandinsky", KandinskyInpaintCombinedPipeline), ("kandinsky22", KandinskyV22InpaintCombinedPipeline),