[Utils] Adds `store()` and `restore()` methods to EMAModel #2302
patil-suraj
merged
16 commits into
main
from
refactor/ema
add store and restore() methods to EMAModel.
2c887e89
patrickvonplaten
approved these changes
on 2023-02-13
patrickvonplaten
commented
on 2023-02-13
Update src/diffusers/training_utils.py
ee5766e2
Merge branch 'main' into refactor/ema
265586fc
Merge branch 'main' into refactor/ema
2cf589d7
Merge branch 'main' into refactor/ema
5f74d2da
make style with doc builder
59869432
williamberman
commented
on 2023-02-15
williamberman
commented
on 2023-02-15
williamberman
approved these changes
on 2023-02-15
remove explicit listing.
db7f0311
williamberman
commented
on 2023-02-16
williamberman
commented
on 2023-02-16
Apply suggestions from code review
4125190e
patrickvonplaten
commented
on 2023-02-16
patrickvonplaten
commented
on 2023-02-16
patrickvonplaten
commented
on 2023-02-16
patrickvonplaten
approved these changes
on 2023-02-16
patil-suraj
commented
on 2023-02-16
Apply suggestions from code review
e6f2e379
Merge branch 'refactor/ema' of https://github.com/huggingface/diffuse…
576adf27
chore: better variable naming.
b3a28c06
better treatment of temp_stored_params
635313ca
make style
9adac8db
remove temporary params from earth 🌎
a2e44779
Merge branch 'main' into refactor/ema
6bc48efc
make fix-copies.
58738646
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub