Sayak Paul 31adeb41cd [Tests] fix sharding tests (#8764) ...

fix sharding tests

2024-07-04 08:50:59 +05:30

..

__init__.py

[Refactor] harmonize the module structure for models in tests (#6738)

2024-02-01 14:23:39 +05:30

test_models_vae_flax.py

[Refactor] harmonize the module structure for models in tests (#6738)

2024-02-01 14:23:39 +05:30

test_models_vae.py

[Tests] fix sharding tests (#8764)

2024-07-04 08:50:59 +05:30

test_models_vq.py

Adding VQGAN Training script (#5483)

2024-05-15 08:47:12 +05:30