Pedro Cuenca 92e1164e2e `mps`: remove warmup passes (#2771) ... * Remove warmup passes in mps tests. * Update mps docs: no warmup pass in PyTorch 2 * Update imports.		2023-03-22 19:29:27 +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.py	[Tests] Correct PT2 (#2724)	2023-03-18 18:38:04 +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_3d_condition.py	[MS Text To Video] Add first text to video (#2738)	2023-03-22 18:39:33 +01:00
test_models_vae.py	`mps`: remove warmup passes (#2771)	2023-03-22 19:29:27 +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	`mps`: remove warmup passes (#2771)	2023-03-22 19:29:27 +01:00