54bc882d96
* 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. |
||
---|---|---|
.. | ||
__init__.py | ||
test_models_unet_1d.py | ||
test_models_unet_2d.py | ||
test_models_unet_2d_condition.py | ||
test_models_unet_2d_flax.py | ||
test_models_vae.py | ||
test_models_vae_flax.py | ||
test_models_vq.py |