Patrick von Platen 9ecd924859 [Tests] Correct PT2 (#2724) ...

* [Tests] Correct PT2

* correct more

* move versatile to nightly

* up

* up

* again

* Apply suggestions from code review

2023-03-18 18:38:04 +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

Adding use_safetensors argument to give more control to users (#2123)

2023-03-16 15:57:43 +01: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_vae_flax.py

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

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

test_models_vae.py

Enabling gradient checkpointing for VAE (#2536)

2023-03-17 14:59:38 -07:00

test_models_vq.py

[Copyright] 2023 (#2524)

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