Suraj Patil bce65cd13a [refactor] make set_attention_slice recursive (#1532) ...

* make attn slice recursive

* remove set_attention_slice from blocks

* fix copies

* make enable_attention_slicing base class method of DiffusionPipeline

* fix set_attention_slice

* fix set_attention_slice

* fix copies

* add tests

* up

* up

* up

* update

* up

* uP

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

2022-12-05 17:31:04 +01:00

..

__init__.py

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

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

test_models_unet_1d.py

Fix dtype model loading (#1449)

2022-11-30 11:31:50 +01:00

test_models_unet_2d_flax.py

Flax support for Stable Diffusion 2 (#1423)

2022-11-29 12:33:21 +01:00

test_models_unet_2d.py

[refactor] make set_attention_slice recursive (#1532)

2022-12-05 17:31:04 +01:00

test_models_vae_flax.py

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

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

test_models_vae.py

[CI] Add slow MPS tests (#1104)

2022-12-05 11:50:24 +01:00

test_models_vq.py

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

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