From 4e698b1088c5ee5588692028803cba12baf4604b Mon Sep 17 00:00:00 2001 From: Edna <88869424+Ednaordinary@users.noreply.github.com> Date: Wed, 11 Jun 2025 18:21:10 -0600 Subject: [PATCH] add chroma to init --- src/diffusers/__init__.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py index f660ab0521..2067e7d9d5 100644 --- a/src/diffusers/__init__.py +++ b/src/diffusers/__init__.py @@ -354,6 +354,7 @@ else: "BlipDiffusionControlNetPipeline", "BlipDiffusionPipeline", "CLIPImageProjection", + "ChromaPipeline", "CogVideoXFunControlPipeline", "CogVideoXImageToVideoPipeline", "CogVideoXPipeline", @@ -769,6 +770,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: AutoencoderTiny, AutoModel, CacheMixin, + ChromaTransformer2DModel, CogVideoXTransformer3DModel, CogView3PlusTransformer2DModel, CogView4Transformer2DModel, @@ -942,6 +944,7 @@ if TYPE_CHECKING or DIFFUSERS_SLOW_IMPORT: AudioLDMPipeline, AuraFlowPipeline, CLIPImageProjection, + ChromaPipeline, CogVideoXFunControlPipeline, CogVideoXImageToVideoPipeline, CogVideoXPipeline,