src/diffusers/schedulers/scheduling_ddpm.py · b6447fa87ef58211146cf922c4a3782dcd6c02e6 · renzhc / diffusers_dcu

Allow DDPM scheduler to use model's predicated variance (#132) · b6447fa8

Eyal Mazuz authored Aug 03, 2022



* Extented the ability of ddpm scheduler
to utilize model that also predict the variance.

* Update src/diffusers/schedulers/scheduling_ddpm.py
Co-authored-by: Anton Lozhkov <aglozhkov@gmail.com>

b6447fa8

scheduling_ddpm.py 7.62 KB

Replace scheduling_ddpm.py