Commit 29d9f02f authored by patil-suraj's avatar patil-suraj
Browse files

BDDMPipeline -> BDDM

parent cdf58a4e
......@@ -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, BDDMPipeline
from .pipelines import DDIM, DDPM, GLIDE, LatentDiffusion, BDDM
from .schedulers import DDIMScheduler, DDPMScheduler, SchedulerMixin
from .schedulers.classifier_free_guidance import ClassifierFreeGuidanceScheduler
......@@ -2,4 +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
from .pipeline_bddm import BDDM
......@@ -262,7 +262,7 @@ class DiffWave(ModelMixin, ConfigMixin):
return self.final_conv(x)
class BDDMPipeline(DiffusionPipeline):
class BDDM(DiffusionPipeline):
def __init__(self, diffwave, noise_scheduler):
super().__init__()
noise_scheduler = noise_scheduler.set_format("pt")
......
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