Fix EMA for multi-gpu training in the unconditional example #1930
improve EMA
a4c91182
style
4e32811a
Merge branch 'main' of https://github.com/huggingface/diffusers into …
4f55f598
one EMA model
2bb2d385
quality
3b274815
fix tests
e9db8cd3
fix test
43f0fe3c
Apply suggestions from code review
db2d359d
re organise the unconditional script
e7ac781f
backwards compatibility
7092f3a8
default to init values for some args
f2b5d406
fix ort script
eeed3af0
issubclass => isinstance
7f84f1be
update state_dict
be021257
docstr
f61474a3
Merge branch 'main' into fix-unconditional-ema
2eb8a0c6
doc
df2d9e0e
pcuenca
approved these changes
on 2023-01-18
Apply suggestions from code review
a94b53b4
use .to if device is passed
c4409e48
deprecate device
bd9f142c
make flake happy
0bafc112
Merge branch 'main' into fix-unconditional-ema
e65e35e7
fix typo
dc1935f3
patil-suraj
deleted the fix-unconditional-ema branch 2 years ago
Login to write a write a comment.
Login via GitHub