diffusers
Add extra performance features for EMAModel, torch._foreach operations and better support for non-blocking CPU offloading
#7685
Merged
Merged
Add extra performance features for EMAModel, torch._foreach operations and better support for non-blocking CPU offloading #7685
sayakpaul
merged
19 commits into
huggingface:main
from
drhead:patch-1
Add support for _foreach operations and non-blocking to EMAModel
f9ead54f
default foreach to false
caab3269
sayakpaul
approved these changes
on 2024-04-16
add non-blocking EMA offloading to SD1.5 T2I example script
49c8b606
fix whitespace
4bbe8232
move foreach to cli argument
527b8cb2
Merge branch 'main' into patch-1
ec887e76
sayakpaul
commented
on 2024-04-23
linting
9d479470
Update README.md re: EMA weight training
a09614e9
Merge branch 'main' into patch-1
ab238daf
sayakpaul
commented
on 2024-04-25
Merge branch 'main' into patch-1
c3268ac0
bghira
commented
on 2024-06-22
correct args.foreach_ema
503a7650
Merge branch 'main' into patch-1
89b07dcd
add tests for foreach ema
662e9580
Merge branch 'main' into patch-1
d3dccda5
code quality
e9ed2845
add foreach to from_pretrained
8bdca089
default foreach false
48039c3e
fix linting
cff1e06a
Merge branch 'main' into patch-1
9c23aedb
Login to write a write a comment.
Login via GitHub