Sayak Paul 7139f0e874 fix: norm group test for UNet3D. (#2959)

2023-04-04 09:01:15 +01:00

..

__init__.py

[Tests] Improve unet / vae tests (#1018)

2022-10-28 13:43:26 +02:00

test_models_unet_1d.py

[Copyright] 2023 (#2524)

2023-03-01 10:31:00 +01:00

test_models_unet_2d_condition.py

Helper function to disable custom attention processors (#2791)

2023-03-27 20:31:19 +02:00

test_models_unet_2d_flax.py

Replace flake8 with ruff and update black (#2279)

2023-02-07 23:46:23 +01:00

test_models_unet_2d.py

[Tests] Correct PT2 (#2724)

2023-03-18 18:38:04 +01:00

test_models_unet_3d_condition.py

fix: norm group test for UNet3D. (#2959)

2023-04-04 09:01:15 +01:00

test_models_vae_flax.py

[Tests] Improve unet / vae tests (#1018)

2022-10-28 13:43:26 +02:00

test_models_vae.py

mps: remove warmup passes (#2771)

2023-03-22 19:29:27 +01:00

test_models_vq.py

mps: remove warmup passes (#2771)

2023-03-22 19:29:27 +01:00