LaurentMazare / diffusers-rs

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

Implement Heun discrete scheduler #37

Closed mspronesti closed 1 year ago

mspronesti commented 1 year ago

This PR aims at integrating Heun Discrete Scheduler into this repository, solving the fourth task metioned in #23 .

This implementation ports all the features supported in the official HF implementation. A couple of observations:

mspronesti commented 1 year ago

One of the workflows (windows-2019 stable) took 40 mins to pass. Not sure if anything weird happened on GitHub side 😅 I recommend re-running it, just to make sure, but I think it was just accidental.

EDIT: I confirm it. I mirrored my branch and run the action. All the checks passed in ~5 mins as expected.

LaurentMazare commented 1 year ago

Thanks for the PR!