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,