Will Berman 0c8fd45894 add only cross attention to simple attention blocks (#3011) ...

* add only cross attention to simple attention blocks

* add test for only_cross_attention re: @patrickvonplaten

* mid_block_only_cross_attention better default

allow mid_block_only_cross_attention to default to
`only_cross_attention` when `only_cross_attention` is given
as a single boolean

2023-05-05 07:22:12 -07:00

..

__init__.py

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

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

test_attention_processor.py

add only cross attention to simple attention blocks (#3011)

2023-05-05 07:22:12 -07:00

test_models_unet_1d.py

Fix config prints and save, load of pipelines (#2849)

2023-05-05 07:22:12 -07: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