Pedro Cuenca b10f527577 Helper function to disable custom attention processors (#2791) ...

* Helper function to disable custom attention processors.

* Restore code deleted by mistake.

* Format

* Fix modeling_text_unet copy.

2023-03-27 20:31:19 +02: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

Skip mps in text-to-video tests (#2792)

2023-03-23 14:39:03 +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