Patrick von Platen 4125756e88 Refactor cross attention and allow mechanism to tweak cross attention function (#1639) ...

* first proposal

* rename

* up

* Apply suggestions from code review

* better

* up

* finish

* up

* rename

* correct versatile

* up

* up

* up

* up

* fix

* Apply suggestions from code review

* make style

* Apply suggestions from code review

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

* add error message

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

2022-12-20 18:49:05 +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 cross attention and allow mechanism to tweak cross attention function (#1639)

2022-12-20 18:49:05 +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