Deprecate `predict_epsilon` #1393
Adapt ddpm, ddpmsolver to prediction_type.
287a5503
Deprecate predict_epsilon in __init__.
74d72731
Bring FlaxDDIMScheduler up to date with DDIMScheduler.
006f8d3c
pcuenca
marked this pull request as draft 3 years ago
Set prediction_type as an ivar for consistency.
0fa4468d
Convert pipeline_ddpm
bfe4e0e7
Adapt tests.
a6709c3e
Adapt unconditional training script.
20462d67
Adapt BitDiffusion example.
e50a2f3e
Add missing kwargs in dpmsolver_multistep
e3adf8c2
Ugly workaround to accept deprecated predict_epsilon when loading
1a19afb6
Merge remote-tracking branch 'origin/main' into deprecate-predict-eps…
bc374f25
pcuenca
marked this pull request as ready for review 3 years ago
Merge remote-tracking branch 'origin/main' into deprecate-predict-eps…
fc2828be
make style
86e7fb03
Remove import no longer in use.
09550733
Apply suggestions from code review
0d3e123f
Use config.prediction_type everywhere
56e9d10d
Add a couple of Flax prediction type tests.
01df570c
make style
4a3cbe2a
Merge branch 'main' into deprecate-predict-epsilon
760f2fb8
fix register deprecated arg
e6b1b29e
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub