20ce68f945
* Add test * up * no bfloat16 for mps * fix * rename test |
||
---|---|---|
.. | ||
__init__.py | ||
test_models_unet_1d.py | ||
test_models_unet_2d.py | ||
test_models_unet_2d_flax.py | ||
test_models_vae.py | ||
test_models_vae_flax.py | ||
test_models_vq.py |