diff --git a/src/diffusers/__init__.py b/src/diffusers/__init__.py index 16f740066e..edf2f1c5ec 100644 --- a/src/diffusers/__init__.py +++ b/src/diffusers/__init__.py @@ -9,6 +9,6 @@ from .models.unet import UNetModel from .models.unet_glide import GLIDESuperResUNetModel, GLIDETextToImageUNetModel from .models.unet_ldm import UNetLDMModel from .pipeline_utils import DiffusionPipeline -from .pipelines import DDIM, DDPM, GLIDE, LatentDiffusion +from .pipelines import DDIM, DDPM, GLIDE, LatentDiffusion, BDDMPipeline from .schedulers import DDIMScheduler, DDPMScheduler, SchedulerMixin from .schedulers.classifier_free_guidance import ClassifierFreeGuidanceScheduler diff --git a/src/diffusers/pipelines/__init__.py b/src/diffusers/pipelines/__init__.py index 18d0e80e1a..288a6878f3 100644 --- a/src/diffusers/pipelines/__init__.py +++ b/src/diffusers/pipelines/__init__.py @@ -2,3 +2,4 @@ from .pipeline_ddim import DDIM from .pipeline_ddpm import DDPM from .pipeline_glide import GLIDE from .pipeline_latent_diffusion import LatentDiffusion +from .pipeline_bddm import BDDMPipeline