From e1ef122260d015255b0a7c075fd08ed114670671 Mon Sep 17 00:00:00 2001 From: Narek Maloyan Date: Mon, 20 Jun 2022 20:11:43 +0000 Subject: [PATCH] fix alphas_cumprod --- src/diffusers/schedulers/scheduling_ddpm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/diffusers/schedulers/scheduling_ddpm.py b/src/diffusers/schedulers/scheduling_ddpm.py index 1831a88bec..bc95c0afa8 100644 --- a/src/diffusers/schedulers/scheduling_ddpm.py +++ b/src/diffusers/schedulers/scheduling_ddpm.py @@ -137,8 +137,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): return pred_prev_sample def forward_step(self, original_sample, noise, t): - sqrt_alpha_prod = self.alpha_prod_t[t] ** 0.5 - sqrt_one_minus_alpha_prod = (1 - self.alpha_prod_t[t]) ** 0.5 + sqrt_alpha_prod = self.alphas_cumprod[t] ** 0.5 + sqrt_one_minus_alpha_prod = (1 - self.alphas_cumprod[t]) ** 0.5 noisy_sample = sqrt_alpha_prod * original_sample + sqrt_one_minus_alpha_prod * noise return noisy_sample