fix(schedulers): bug when predict_type is sample

LaurentMazare / diffusers-rs

An implementation of the diffusers api in Rust

Apache License 2.0

535 stars 55 forks source link

fix(schedulers): bug when predict_type is sample #57

Closed mspronesti closed 1 year ago

mspronesti commented 1 year ago

Hi, this PR aims at solving solving a minor bug in DDIM scheduler when prediction type is Sample (as in 2094) and to align add_noise prototype to all other schedulers (and to HF diffusers).

LaurentMazare commented 1 year ago

Thanks!