-
Notifications
You must be signed in to change notification settings - Fork 6.5k
Add Singlestep DPM-Solver (singlestep high-order schedulers) #1442
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
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.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
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 <[email protected]>
…face#1442) * add singlestep dpmsolver * fix a style typo * fix a style typo * add docs * finish Co-authored-by: Patrick von Platen <[email protected]>
…face#1442) * add singlestep dpmsolver * fix a style typo * fix a style typo * add docs * finish Co-authored-by: Patrick von Platen <[email protected]>




#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.