-
Notifications
You must be signed in to change notification settings - Fork 6.5k
support v prediction in other schedulers #1505
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
Conversation
|
The documentation is not available anymore as the PR was closed or merged. |
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.
Could we use this PR to also add tests to all the schedulers?
|
The simple test I had from the original v-prediction pr I can add ddpm to this or in another pr if it's not done yet. |
|
Or just copy the code from #818 for ddpm |
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.
Nice!
* support v prediction in other schedulers * v heun * add tests for v pred * fix tests * fix test euler a * v ddpm
* support v prediction in other schedulers * v heun * add tests for v pred * fix tests * fix test euler a * v ddpm
|
Hi, is there a documentation somewhere describing what it is / how to use v-prediction with the schedulers? Or is this automatically used? |
|
It's automatically used based on the value in the You can also pass |
* support v prediction in other schedulers * v heun * add tests for v pred * fix tests * fix test euler a * v ddpm
This PR adds v-prediction support to
PNDMScheduler,LMSDiscreteSchedulerandEulerAncestralDiscreteScheduler.