diffusers
2ada094b - Add extra performance features for EMAModel, torch._foreach operations and better support for non-blocking CPU offloading (#7685)

Commit
1 year ago
Add extra performance features for EMAModel, torch._foreach operations and better support for non-blocking CPU offloading (#7685) * Add support for _foreach operations and non-blocking to EMAModel * default foreach to false * add non-blocking EMA offloading to SD1.5 T2I example script * fix whitespace * move foreach to cli argument * linting * Update README.md re: EMA weight training * correct args.foreach_ema * add tests for foreach ema * code quality * add foreach to from_pretrained * default foreach false * fix linting --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: drhead <a@a.a>
Author
Parents
Loading