diff --git a/src/diffusers/pipelines/__init__.py b/src/diffusers/pipelines/__init__.py index b4c5d442e1..0f585b5de9 100644 --- a/src/diffusers/pipelines/__init__.py +++ b/src/diffusers/pipelines/__init__.py @@ -562,6 +562,8 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: PaintByExamplePipeline, PIAPipeline, SemanticStableDiffusionPipeline, + ShapEImg2ImgPipeline, + ShapEPipeline, StableDiffusionAttendAndExcitePipeline, StableDiffusionControlNetXSPipeline, StableDiffusionDiffEditPipeline,