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_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_unet_2d.py

[LoRA] Add LoRA training script (#1884)

2023-01-18 18:05:51 +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

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +01:00

test_models_vq.py

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

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