[Tests] reduce block sizes of UNet and VAE tests #7560
sayakpaul
merged
33 commits into
main
from
reduce-block-sizes-unet
reduce block sizes for unet1d.
a10e5b36
reduce blocks for unet_2d.
df00930c
reduce block size for unet_motion
247bc88e
increase channels.
44ba5946
correctly increase channels.
5834d192
reduce number of layers in unet2dconditionmodel tests.
fcc41be7
reduce block sizes for unet2dconditionmodel tests
2148b421
reduce block sizes for unet3dconditionmodel.
c1dfa1f3
fix: test_feed_forward_chunking
cb7ad38d
fix: test_forward_with_norm_groups
6059849a
skip spatiotemporal tests on MPS.
2fec1315
reduce block size in AutoencoderKL.
6fddb91b
reduce block sizes for vqmodel.
57250d83
further reduce block size.
6fe63267
make style.
694fda19
sayakpaul
commented
on 2024-04-02
Empty-Commit
36f365fe
Merge branch 'main' into reduce-block-sizes-unet
8441f313
DN6
commented
on 2024-04-03
DN6
commented
on 2024-04-03
DN6
commented
on 2024-04-03
DN6
commented
on 2024-04-03
Merge branch 'main' into reduce-block-sizes-unet
edbf0ff3
reduce sizes for ConsistencyDecoderVAETests
0e4c9625
further reduction.
2289d5a3
further block reductions in AutoencoderKL and AssymetricAutoencoderKL.
adbad35c
massively reduce the block size in unet2dcontionmodel.
8d09df46
reduce sizes for unet3d
0ae049b2
fix tests in unet3d.
cc2f3dd2
reduce blocks further in motion unet.
d62d0ee7
fix: output shape
9ce9035e
add attention_head_dim to the test configuration.
d7572b6f
remove unexpected keyword arg
9f9532d9
up a bit.
7d00d064
groups.
5fe6974b
up again
4171cd6d
fix
e7e7209a
Merge branch 'main' into reduce-block-sizes-unet
910f6995
DN6
approved these changes
on 2024-04-05
Login to write a write a comment.
Login via GitHub