Sayak Paul 1c60e094de [Tests] reduce block sizes of UNet and VAE tests (#7560) ...

* reduce block sizes for unet1d.

* reduce blocks for unet_2d.

* reduce block size for unet_motion

* increase channels.

* correctly increase channels.

* reduce number of layers in unet2dconditionmodel tests.

* reduce block sizes for unet2dconditionmodel tests

* reduce block sizes for unet3dconditionmodel.

* fix: test_feed_forward_chunking

* fix: test_forward_with_norm_groups

* skip spatiotemporal tests on MPS.

* reduce block size in AutoencoderKL.

* reduce block sizes for vqmodel.

* further reduce block size.

* make style.

* Empty-Commit

* reduce sizes for ConsistencyDecoderVAETests

* further reduction.

* further block reductions in AutoencoderKL and AssymetricAutoencoderKL.

* massively reduce the block size in unet2dcontionmodel.

* reduce sizes for unet3d

* fix tests in unet3d.

* reduce blocks further in motion unet.

* fix: output shape

* add attention_head_dim to the test configuration.

* remove unexpected keyword arg

* up a bit.

* groups.

* up again

* fix

2024-04-05 10:08:32 +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] reduce block sizes of UNet and VAE tests (#7560)

2024-04-05 10:08:32 +05:30

test_models_vq.py

[Tests] reduce block sizes of UNet and VAE tests (#7560)

2024-04-05 10:08:32 +05:30