LaurentMazare / diffusers-rs

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

Support DPM MultiStep Scheduler #17

Closed ajmwagar closed 1 year ago

ajmwagar commented 1 year ago

Would be great to have support for the DPM MultiStep Scheduler.

Diffusers recommends using it as it's the fastest scheduler at the moment: https://huggingface.co/docs/diffusers/v0.9.0/en/api/pipelines/stable_diffusion_2#available-checkpoints

Link to diffusers python implementation:

https://github.com/huggingface/diffusers/blob/v0.9.0/src/diffusers/schedulers/scheduling_dpmsolver_multistep.py#L57

mspronesti commented 1 year ago

I'm working on it. I will open a PR soon:)

mspronesti commented 1 year ago

I guess this issue can be closed now.

LaurentMazare commented 1 year ago

As mentioned by @mspronesti this has been added (as well as many other schedulers :) ).