LaurentMazare / diffusers-rs

An implementation of the diffusers api in Rust
Apache License 2.0
539 stars 55 forks source link

Implement PNDM scheduler #41

Closed mspronesti closed 1 year ago

mspronesti commented 1 year ago

Hi @LaurentMazare, this PR aims at integrating the PNDM scheduler into this repository, addressing the 7th task mentioned in #23 .

This implementation only ports the relevant features for stable diffusion, i.e. only the linear multi-step propagation method, skipping the Runge-Kutta steps. In other words, it doesn't port this method. I'm planning to add it in a feature PR if we eventually support new pipelines, unless you prefer to have the full scheduler now. If it is the case, please let me know and I will append a new commit to this PR :)

LaurentMazare commented 1 year ago

Thanks!