Use DDPMScheduler instead of DDIMScheduler for noise_scheduler since PR

huggingface#6305 has been merged.
dg845 · Dec 26, 2023 · 60a9ea7 · 60a9ea7
1 parent 100ef46
commit 60a9ea7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/examples/add/train_add_distill_sd_wds.py b/examples/add/train_add_distill_sd_wds.py
@@ -57,7 +57,7 @@
 import diffusers
 from diffusers import (
     AutoencoderKL,
-    DDIMScheduler,
+    DDPMScheduler,
     StableDiffusionPipeline,
     UNet2DConditionModel,
 )
@@ -1119,12 +1119,12 @@ def main(args):
 
     # 1. Create the noise scheduler and the desired noise schedule.
     # Enforce zero terminal SNR (see section 3.1 of ADD paper)
-    teacher_scheduler = DDIMScheduler.from_pretrained(
+    teacher_scheduler = DDPMScheduler.from_pretrained(
         args.pretrained_teacher_model, subfolder="scheduler", revision=args.teacher_revision
     )
     if not teacher_scheduler.config.rescale_betas_zero_snr:
         teacher_scheduler.config["rescale_betas_zero_snr"] = True
-    noise_scheduler = DDIMScheduler(**teacher_scheduler.config)
+    noise_scheduler = DDPMScheduler(**teacher_scheduler.config)
 
     # DDIMScheduler calculates the alpha and sigma noise schedules (based on the alpha bars) for us
     alpha_schedule = torch.sqrt(noise_scheduler.alphas_cumprod)

diff --git a/examples/add/train_add_distill_sdxl_wds.py b/examples/add/train_add_distill_sdxl_wds.py
@@ -57,7 +57,7 @@
 import diffusers
 from diffusers import (
     AutoencoderKL,
-    DDIMScheduler,
+    DDPMScheduler,
     StableDiffusionXLPipeline,
     UNet2DConditionModel,
 )
@@ -1177,12 +1177,12 @@ def main(args):
 
     # 1. Create the noise scheduler and the desired noise schedule.
     # Enforce zero terminal SNR (see section 3.1 of ADD paper)
-    teacher_scheduler = DDIMScheduler.from_pretrained(
+    teacher_scheduler = DDPMScheduler.from_pretrained(
         args.pretrained_teacher_model, subfolder="scheduler", revision=args.teacher_revision
     )
     if not teacher_scheduler.config.rescale_betas_zero_snr:
         teacher_scheduler.config["rescale_betas_zero_snr"] = True
-    noise_scheduler = DDIMScheduler(**teacher_scheduler.config)
+    noise_scheduler = DDPMScheduler(**teacher_scheduler.config)
 
     # DDPMScheduler calculates the alpha and sigma noise schedules (based on the alpha bars) for us
     alpha_schedule = torch.sqrt(noise_scheduler.alphas_cumprod)