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 ...@@ -9,6 +9,6 @@ from .models.unet import UNetModel
from .models.unet_glide import GLIDESuperResUNetModel, GLIDETextToImageUNetModel from .models.unet_glide import GLIDESuperResUNetModel, GLIDETextToImageUNetModel
from .models.unet_ldm import UNetLDMModel from .models.unet_ldm import UNetLDMModel
from .pipeline_utils import DiffusionPipeline 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 import DDIMScheduler, DDPMScheduler, SchedulerMixin
from .schedulers.classifier_free_guidance import ClassifierFreeGuidanceScheduler from .schedulers.classifier_free_guidance import ClassifierFreeGuidanceScheduler
...@@ -2,4 +2,4 @@ from .pipeline_ddim import DDIM ...@@ -2,4 +2,4 @@ from .pipeline_ddim import DDIM
from .pipeline_ddpm import DDPM from .pipeline_ddpm import DDPM
from .pipeline_glide import GLIDE from .pipeline_glide import GLIDE
from .pipeline_latent_diffusion import LatentDiffusion from .pipeline_latent_diffusion import LatentDiffusion
from .pipeline_bddm import BDDMPipeline from .pipeline_bddm import BDDM
...@@ -262,7 +262,7 @@ class DiffWave(ModelMixin, ConfigMixin): ...@@ -262,7 +262,7 @@ class DiffWave(ModelMixin, ConfigMixin):
return self.final_conv(x) return self.final_conv(x)
class BDDMPipeline(DiffusionPipeline): class BDDM(DiffusionPipeline):
def __init__(self, diffwave, noise_scheduler): def __init__(self, diffwave, noise_scheduler):
super().__init__() super().__init__()
noise_scheduler = noise_scheduler.set_format("pt") 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