feat: add prediction argument #334
Open
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
daniandtheweb
force-pushed
the
prediction_argument
branch
2 times, most recently
from
e995b3b
to
1b00b0c
Compare
daniandtheweb
force-pushed
the
prediction_argument
branch
2 times, most recently
from
24c784b
to
b9064ac
Compare
Green-Sky
approved these changes
Aug 25, 2024
daniandtheweb
force-pushed
the
prediction_argument
branch
from
b9064ac
to
3beb06f
Compare
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR adresses the issue #325 by adding an argument to force a specific prediction mode.
Some experimental models (such as EasyFluff and similar models) may introduce different prediction modes from what's expected (Stable Diffusion 1.5 uses EPS but in this case the mentioned model uses v-predict) and this change allows using them correctly by manually adjusting the prediction type.