diff --git a/src/diffusers/pipelines/__init__.py b/src/diffusers/pipelines/__init__.py index 268e5c2a8c..d20d609ff9 100644 --- a/src/diffusers/pipelines/__init__.py +++ b/src/diffusers/pipelines/__init__.py @@ -148,6 +148,7 @@ else: "AudioLDM2UNet2DConditionModel", ] _import_structure["blip_diffusion"] = ["BlipDiffusionPipeline"] + _import_structure["chroma"] = ["ChromaPipeline"] _import_structure["cogvideo"] = [ "CogVideoXPipeline", "CogVideoXImageToVideoPipeline",