Anton Lozhkov 7c82a16fc1 Fix EMA for multi-gpu training in the unconditional example (#1930) ...

* improve EMA

* style

* one EMA model

* quality

* fix tests

* fix test

* Apply suggestions from code review

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* re organise the unconditional script

* backwards compatibility

* default to init values for some args

* fix ort script

* issubclass => isinstance

* update state_dict

* docstr

* doc

* Apply suggestions from code review

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* use .to if device is passed

* deprecate device

* make flake happy

* fix typo

Co-authored-by: patil-suraj <surajp815@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2023-01-19 11:35:55 +01:00

..

fixtures/custom_pipeline

Make repo structure consistent (#1862)

2022-12-30 11:51:08 +01:00

models

[LoRA] Add LoRA training script (#1884)

2023-01-18 18:05:51 +01:00

pipelines

[Dit] Fix dit tests (#2034)

2023-01-19 01:50:22 +01:00

repo_utils

Move accelerate to a soft-dependency (#1134)

2022-11-04 14:58:52 +01:00

__init__.py

fix issues with loading, add test for pipeline

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

conftest.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_config.py

Bump to 0.12.0.dev0 (#1771)

2022-12-19 18:44:08 +01:00

test_layers_utils.py

Test ResnetBlock2D (#1850)

2023-01-04 22:57:32 +01:00

test_modeling_common_flax.py

Allow to set config params directly in init (#1419)

2022-11-25 15:07:09 +01:00

test_modeling_common.py

Fix EMA for multi-gpu training in the unconditional example (#1930)

2023-01-19 11:35:55 +01:00

test_outputs.py

Fix BaseOutput initialization from dict (#570)

2022-09-20 18:32:16 +02:00

test_pipelines_common.py

DiT Pipeline (#1806)

2023-01-17 23:09:29 +01:00

test_pipelines_flax.py

Make height and width optional (#1401)

2022-11-24 18:23:59 +01:00

test_pipelines_onnx_common.py

Reorganize pipeline tests (#963)

2022-10-24 16:34:01 +02:00

test_pipelines.py

Allow converting Flax to PyTorch by adding a "from_flax" keyword (#1900)

2023-01-12 20:00:35 +01:00

test_scheduler_flax.py

[Flax] Stateless schedulers, fixes and refactors (#1661)

2022-12-20 01:42:41 +01:00

test_scheduler.py

[Black] Update black library (#2007)

2023-01-16 15:16:28 +01:00

test_training.py

[Utils] Add deprecate function and move testing_utils under utils (#659)

2022-10-03 23:44:24 +02:00

test_unet_2d_blocks.py

Add tests for 2D UNet blocks (#1945)

2023-01-16 12:53:05 +01:00

test_unet_blocks_common.py

Add tests for 2D UNet blocks (#1945)

2023-01-16 12:53:05 +01:00

test_utils.py

[Deprecate] Correct stacklevel (#1483)

2022-12-01 16:28:10 +01:00