Patrick von Platen 3267649480 Make sure VAE attention works with Torch 2_0 (#3200) ...

* Make sure attention works with Torch 2_0

* make style

* Fix more

2023-05-05 07:22:14 -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_layers_utils.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_lora_layers.py

make style

2023-05-05 07:22:14 -07:00

test_modeling_common_flax.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_modeling_common.py

Fix config deprecation (#3129)

2023-05-05 07:22:13 -07:00

test_models_unet_1d.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_models_unet_2d_condition.py

adding custom diffusion training to diffusers examples (#3031)

2023-05-05 07:22:14 -07: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] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_models_unet_3d_condition.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_models_vae_flax.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_models_vae.py

Make sure VAE attention works with Torch 2_0 (#3200)

2023-05-05 07:22:14 -07:00

test_models_vq.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_unet_2d_blocks.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00

test_unet_blocks_common.py

[Tests] parallelize (#3078)

2023-05-05 07:22:13 -07:00