Commit 86da45bc authored by patil-suraj's avatar patil-suraj
Browse files

add BDDMPipeline in init

parent 99e6d64f
......@@ -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
......@@ -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
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment