Patrick von Platen 20ce68f945 Fix dtype model loading (#1449) ...

* Add test

* up

* no bfloat16 for mps

* fix

* rename test

2022-11-30 11:31:50 +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

Flax support for Stable Diffusion 2 (#1423)

2022-11-29 12:33:21 +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

default fast model loading 🔥 (#1115)

2022-11-03 17:25:57 +01:00

test_models_vq.py

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

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