-
Notifications
You must be signed in to change notification settings - Fork 34
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
[Feature Request] Support v-prediction #25
Comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Some SD finetunes (and the official Stable Diffusion 2) use v-prediction because they observe that it's better
Currently, v-prediction models will export with Olive and run fine with the ONNX diffusers pipeline, but not here. This is "a castle" on one I exported
I tried modifying
DpmPlusPlus2MScheduler::ApplyStep
by modifying thepredictedOriginalSample
based off one of the diffusers implementationsI could only make the result semi-coherentIn
DPMPlusPlus2MScheduler
, the formula from HF's euler discrete worksIf you could finish this
(haven't figured out(I have everything here), it would greatly improve compatibility. I uploaded the model in case it's useful.EulerAncestralScheduler
)The text was updated successfully, but these errors were encountered: