Skip to content

GitLab

Commit e1ef1222 authored by Narek Maloyan's avatar Narek Maloyan
Browse files

fix alphas_cumprod

parent 4497e78d
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
src/diffusers/schedulers/scheduling_ddpm.py
View file @ e1ef1222
...@@ -137,8 +137,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): ...@@ -137,8 +137,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
return pred_prev_sample return pred_prev_sample
def forward_step(self, original_sample, noise, t): def forward_step(self, original_sample, noise, t):
sqrt_alpha_prod = self.alpha_prod_t[t] ** 0.5 sqrt_alpha_prod = self.alphas_cumprod[t] ** 0.5
sqrt_one_minus_alpha_prod = (1 - self.alpha_prod_t[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 noisy_sample = sqrt_alpha_prod * original_sample + sqrt_one_minus_alpha_prod * noise
return noisy_sample return noisy_sample
......
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