Patrick von Platen cd91fc06fe Re-add xformers enable to UNet2DCondition (#1627) ...

* finish

* fix

* Update tests/models/test_models_unet_2d.py

* style

Co-authored-by: Anton Lozhkov <anton@huggingface.co>

2022-12-09 14:05:38 +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

Re-add xformers enable to UNet2DCondition (#1627)

2022-12-09 14:05:38 +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