f07a16e09b
* boom boom * remove duplicate arg * add use_linear_proj arg * fix copies * style * add fast tests * use_linear_proj -> use_linear_projection |
||
---|---|---|
.. | ||
__init__.py | ||
test_models_unet_1d.py | ||
test_models_unet_2d.py | ||
test_models_vae.py | ||
test_models_vae_flax.py | ||
test_models_vq.py |