diffusers/tests/models
Pedro Cuenca 54bc882d96
`mps` test fixes (#2470)
* Skip variant tests (UNet1d, UNetRL) on mps.

mish op not yet supported.

* Exclude a couple of panorama tests on mps

They are too slow for fast CI.

* Exclude mps panorama from more tests.

* mps: exclude all fast panorama tests as they keep failing.
2023-02-24 15:19:53 +01:00
..
__init__.py [Tests] Improve unet / vae tests (#1018) 2022-10-28 13:43:26 +02:00
test_models_unet_1d.py `mps` test fixes (#2470) 2023-02-24 15:19:53 +01:00
test_models_unet_2d.py [LoRA] Add LoRA training script (#1884) 2023-01-18 18:05:51 +01:00
test_models_unet_2d_condition.py Fix running LoRA with xformers (#2286) 2023-02-13 11:58:18 +01:00
test_models_unet_2d_flax.py Replace flake8 with ruff and update black (#2279) 2023-02-07 23:46:23 +01:00
test_models_vae.py Replace flake8 with ruff and update black (#2279) 2023-02-07 23:46:23 +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