1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

remove extra min arg @sayakpaul

This commit is contained in:
William Berman
2023-04-07 11:48:16 -07:00
committed by Will Berman
parent 1dc856e508
commit 1875c35aeb

View File

@@ -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