From 0cbefefac3363666ea2f1b1f730a019214a8b3d4 Mon Sep 17 00:00:00 2001 From: William Berman Date: Fri, 7 Apr 2023 11:49:53 -0700 Subject: [PATCH] clamp comment @sayakpaul --- src/diffusers/schedulers/scheduling_ddpm.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/diffusers/schedulers/scheduling_ddpm.py b/src/diffusers/schedulers/scheduling_ddpm.py index b78a003e2b..59db976620 100644 --- a/src/diffusers/schedulers/scheduling_ddpm.py +++ b/src/diffusers/schedulers/scheduling_ddpm.py @@ -214,6 +214,8 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): # and sample from it to get previous sample # x_{t-1} ~ N(pred_prev_sample, variance) == add variance to pred_sample variance = (1 - alpha_prod_t_prev) / (1 - alpha_prod_t) * current_beta_t + + # we always take the log of variance, so clamp it to ensure it's not 0 variance = torch.clamp(variance, min=1e-20) if variance_type is None: