diffusers

mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

Andy 116f70cbf8 Enabling gradient checkpointing for VAE (#2536 )

* updated black format

* update black format

* make style format

* updated line endings

* update code formatting

* Update examples/research_projects/onnxruntime/text_to_image/train_text_to_image.py

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* Update src/diffusers/models/vae.py

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* Update src/diffusers/models/vae.py

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

* added vae gradient checkpointing test

* make style

---------

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Will Berman <wlbberman@gmail.com>

2023-03-17 14:59:38 -07:00

__init__.py

[Tests] Improve unet / vae tests (#1018 )

2022-10-28 13:43:26 +02:00

test_models_unet_1d.py

[Copyright] 2023 (#2524 )

2023-03-01 10:31:00 +01:00

test_models_unet_2d_condition.py

Adding use_safetensors argument to give more control to users (#2123 )

2023-03-16 15:57:43 +01:00

test_models_unet_2d_flax.py

Replace flake8 with ruff and update black (#2279 )