drhead 2ada094bff 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>

2024-06-24 14:03:47 +05:30

..

test_check_copies.py

#7535 Update FloatTensor type hints to Tensor (#7883)

2024-05-10 09:53:31 -10:00

test_check_dummies.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_config.py

Update loading of config from a file in test_config.py (#7344)

2024-03-18 11:47:36 -10:00

test_dependencies.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_ema.py

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

2024-06-24 14:03:47 +05:30

test_hub_utils.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_image_processor.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_outputs.py

Register BaseOutput subclasses as supported torch.utils._pytree nodes (#5459)

2023-10-24 15:01:47 +05:30

test_training.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_utils.py

Remove distutils (#7455)

2024-03-25 06:44:53 +05:30

test_video_processor.py

[Core] introduce videoprocessor. (#7776)

2024-05-10 21:02:36 +02:00