From 1875c35aebaf98a435df8eda35b28cb3c0eda17e Mon Sep 17 00:00:00 2001 From: William Berman Date: Fri, 7 Apr 2023 11:48:16 -0700 Subject: [PATCH] remove extra min arg @sayakpaul --- src/diffusers/schedulers/scheduling_ddpm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/schedulers/scheduling_ddpm.py b/src/diffusers/schedulers/scheduling_ddpm.py index 481010fcb7..b78a003e2b 100644 --- a/src/diffusers/schedulers/scheduling_ddpm.py +++ b/src/diffusers/schedulers/scheduling_ddpm.py @@ -224,7 +224,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): variance = variance # for rl-diffuser https://arxiv.org/abs/2205.09991 elif variance_type == "fixed_small_log": - variance = torch.log(variance, min=1e-20) + variance = torch.log(variance) variance = torch.exp(0.5 * variance) elif variance_type == "fixed_large": variance = current_beta_t