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

..

fixtures

change to 2024 in the license (#6902)

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

lora

[SD3 LoRA] Fix list index out of range (#8584)

2024-06-21 21:17:34 +05:30

models

a few fix for shard checkpoints (#8656)

2024-06-21 12:50:58 +05:30

others

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

pipelines

[LoRA] get rid of the legacy lora remnants and make our codebase lighter (#8623)

2024-06-21 16:36:05 +05:30

schedulers

[Flax tests] reduce tolerance for a flax test (#8640)

2024-06-20 00:48:08 +04:00

single_file

[tests] decorate StableDiffusion21PipelineSingleFileSlowTests with slow. (#7941)

2024-05-14 14:26:21 -10:00

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

conftest.py

change to 2024 in the license (#6902)

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