Will Berman 41ae670828 move activation dispatches into helper function (#3656) ...

* move activation dispatches into helper function

* tests

2023-06-05 12:30:48 -07:00

..

__init__.py

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

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

test_activations.py

move activation dispatches into helper function (#3656)

2023-06-05 12:30:48 -07:00

test_attention_processor.py

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

2023-04-11 14:38:50 -07:00

test_layers_utils.py

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_lora_layers.py

Support Kohya-ss style LoRA file format (in a limited capacity) (#3437)

2023-06-02 17:40:24 +05:30

test_modeling_common_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_modeling_common.py

Run torch.compile tests in separate subprocesses (#3503)

2023-05-23 19:24:17 +02:00

test_models_unet_1d.py

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_models_unet_2d_condition.py

Support for cross-attention bias / mask (#2634)

2023-05-22 17:27:15 +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

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_models_unet_3d_condition.py

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_models_vae_flax.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_models_vae.py

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_models_vq.py

Refactor full determinism (#3485)

2023-05-22 11:15:11 +01:00

test_unet_2d_blocks.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00

test_unet_blocks_common.py

[Tests] parallelize (#3078)

2023-04-13 13:32:57 +01:00