Commit 1875c35a authored by William Berman's avatar William Berman Committed by Will Berman
Browse files

remove extra min arg @sayakpaul

parent 1dc856e5
...@@ -224,7 +224,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin): ...@@ -224,7 +224,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
variance = variance variance = variance
# for rl-diffuser https://arxiv.org/abs/2205.09991 # for rl-diffuser https://arxiv.org/abs/2205.09991
elif variance_type == "fixed_small_log": elif variance_type == "fixed_small_log":
variance = torch.log(variance, min=1e-20) variance = torch.log(variance)
variance = torch.exp(0.5 * variance) variance = torch.exp(0.5 * variance)
elif variance_type == "fixed_large": elif variance_type == "fixed_large":
variance = current_beta_t variance = current_beta_t
......
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