From da7d4cf2005df264e684976aadbc96fa38f0b53b Mon Sep 17 00:00:00 2001 From: nicolas-dufour <33259879+nicolas-dufour@users.noreply.github.com> Date: Mon, 29 Aug 2022 15:40:49 +0100 Subject: [PATCH] [BugFix]: Fixed add_noise in LMSDiscreteScheduler (#253) * Fixed add_noise in LMSDiscreteScheduler * Linting * Update src/diffusers/schedulers/scheduling_lms_discrete.py Co-authored-by: Anton Lozhkov Co-authored-by: Anton Lozhkov --- src/diffusers/schedulers/scheduling_lms_discrete.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/diffusers/schedulers/scheduling_lms_discrete.py b/src/diffusers/schedulers/scheduling_lms_discrete.py index 55dd3dbec8..9ac39c79d5 100644 --- a/src/diffusers/schedulers/scheduling_lms_discrete.py +++ b/src/diffusers/schedulers/scheduling_lms_discrete.py @@ -124,10 +124,8 @@ class LMSDiscreteScheduler(SchedulerMixin, ConfigMixin): return {"prev_sample": prev_sample} def add_noise(self, original_samples, noise, timesteps): - alpha_prod = self.alphas_cumprod[timesteps] - alpha_prod = self.match_shape(alpha_prod, original_samples) - - noisy_samples = (alpha_prod**0.5) * original_samples + ((1 - alpha_prod) ** 0.5) * noise + sigmas = self.match_shape(self.sigmas, noise) + noisy_samples = original_samples + noise * sigmas[timesteps] return noisy_samples def __len__(self):