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.py	default fast model loading 🔥 (#1115)	2022-11-03 17:25:57 +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