dg845 4c8a05f115 Fix Consistency Models UNet2DMidBlock2D Attention GroupNorm Bug (#4863) ...

* Add attn_groups argument to UNet2DMidBlock2D to control theinternal Attention block's GroupNorm.

* Add docstring for attn_norm_num_groups in UNet2DModel.

* Since the test UNet config uses resnet_time_scale_shift == 'scale_shift', also set attn_norm_num_groups to 32.

* Add test for attn_norm_num_groups to UNet2DModelTests.

* Fix expected slices for slow tests.

* Also fix tolerances for slow tests.

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2023-09-15 11:27:51 +01: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

relax tol attention conversion test (#3842)

2023-06-21 12:35:38 -07:00

test_layers_utils.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_lora_layers.py

[Lora] fix lora fuse unfuse (#5003)

2023-09-13 11:21:04 +02:00

test_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_models_prior.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_models_unet_1d.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_models_unet_2d_condition.py

fix custom diffusion tests (#4996)

2023-09-12 17:50:47 +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

Fix Consistency Models UNet2DMidBlock2D Attention GroupNorm Bug (#4863)

2023-09-15 11:27:51 +01:00

test_models_unet_3d_condition.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_models_vae_flax.py

[Tests] parallelize (#3078)

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

test_models_vae.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_models_vq.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_unet_2d_blocks.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_unet_blocks_common.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00