Pedro Cuenca 813744e5f3 MPS schedulers: don't use float64 (#1169) ...

* Schedulers: don't use float64 on mps

* Test set_timesteps() on device (float schedulers).

* SD pipeline: use device in set_timesteps.

* SD in-painting pipeline: use device in set_timesteps.

* Tests: fix mps crashes.

* Skip test_load_pipeline_from_git on mps.

Not compatible with float16.

* Use device.type instead of str in Euler schedulers.

2022-11-08 13:11:33 +01:00

..

__init__.py

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

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

test_models_unet_1d.py

default fast model loading 🔥 (#1115)

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

test_models_unet_2d.py

MPS schedulers: don't use float64 (#1169)

2022-11-08 13:11:33 +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