Pedro Cuenca aa0531fa8d Skip mps in text-to-video tests (#2792) ...

* Skip mps in text-to-video tests.

* style

* Skip UNet3D mps tests.

2023-03-23 14:39:03 +01:00

..

__init__.py

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

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

test_models_unet_1d.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_models_unet_2d_condition.py

Fix typos (#2715)

2023-03-21 13:45:04 +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

[Tests] Correct PT2 (#2724)

2023-03-18 18:38:04 +01:00

test_models_unet_3d_condition.py

Skip mps in text-to-video tests (#2792)

2023-03-23 14:39:03 +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

mps: remove warmup passes (#2771)

2023-03-22 19:29:27 +01:00

test_models_vq.py

mps: remove warmup passes (#2771)

2023-03-22 19:29:27 +01:00