search

LaurentMazare / diffusers-rs

An implementation of the diffusers api in Rust
Apache License 2.0
521 stars 54 forks source link

Implement thresholding for DPM Solver Multistep scheduler #43

Closed mspronesti closed 1 year ago

mspronesti commented 1 year ago

Hi @LaurentMazare , this PR aims at supporting thresholding in the DPM Solver Multistep scheduler, thus completing #17. Please notice that, as declared in the original paper and in diffuser's documentation, thresholding is unsuitable for latent-space diffusion models such as stable-diffusion. Therefore, this feature is supposed to be helpful when diffusers-rs will support other pipelines.

LaurentMazare commented 1 year ago

Thanks!