fixed typo in comment
This commit is contained in:
parent
bd9c9fbfbe
commit
c3c1bdf8e2
|
@ -92,9 +92,9 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
|
||||||
alpha_prod_t = self.alphas_cumprod[t]
|
alpha_prod_t = self.alphas_cumprod[t]
|
||||||
alpha_prod_t_prev = self.alphas_cumprod[t - 1] if t > 0 else self.one
|
alpha_prod_t_prev = self.alphas_cumprod[t - 1] if t > 0 else self.one
|
||||||
|
|
||||||
# For t > 0, compute predicted variance βt (see formala (6) and (7) from https://arxiv.org/pdf/2006.11239.pdf)
|
# For t > 0, compute predicted variance βt (see formula (6) and (7) from https://arxiv.org/pdf/2006.11239.pdf)
|
||||||
# and sample from it to get previous sample
|
# and sample from it to get previous sample
|
||||||
# x_{t-1} ~ N(pred_prev_sample, variance) == add variane to pred_sample
|
# x_{t-1} ~ N(pred_prev_sample, variance) == add variance to pred_sample
|
||||||
variance = (1 - alpha_prod_t_prev) / (1 - alpha_prod_t) * self.betas[t]
|
variance = (1 - alpha_prod_t_prev) / (1 - alpha_prod_t) * self.betas[t]
|
||||||
|
|
||||||
if variance_type is None:
|
if variance_type is None:
|
||||||
|
|
Loading…
Reference in New Issue