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.py	Refactor cross attention and allow mechanism to tweak cross attention function (#1639)	2022-12-20 18:49:05 +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_vae.py	[CI] Add slow MPS tests (#1104)	2022-12-05 11:50:24 +01:00
test_models_vae_flax.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00
test_models_vq.py	[Tests] Improve unet / vae tests (#1018)	2022-10-28 13:43:26 +02:00