Repository navigation
Add Singlestep DPM-Solver (singlestep high-order schedulers) - #1442
Conversation
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. |
|
Emmm, I don't know what happened with the tests... |
|
Here are some of my own suggestions:
Nevertheless, the difference between multi-step DPM-Solver++ and single-step DPM-Solver++ on stable-diffusion seems very small. Overall, I don't have much experience with which one is better, and I hope the community can have some more findings. |
patrickvonplaten
left a comment
There was a problem hiding this comment.
This looks great to me @LuChengTHU - thanks a lot!
Could you maybe add some quick tests to tests/test_scheduler.py ?
|
Implementation is exactly in line with the current design so that's perfect :-) |
|
No problem. I will add tests and flax version implementations :) |
|
Cool! Let me know if you need any help :-) |
…face#1442) * add singlestep dpmsolver * fix a style typo * fix a style typo * add docs * finish Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
…face#1442) * add singlestep dpmsolver * fix a style typo * fix a style typo * add docs * finish Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
…face#1442) * add singlestep dpmsolver * fix a style typo * fix a style typo * add docs * finish Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>




#1308
I've implemented the single-step high-order solvers in an elegant way. It is very similar to the implementations with the multi-step schedulers, and we do not need to modify pipelines.
If this implementation is fine for you, I will further add the tests and the flax version.