Vishnu V Jaddipal 8dba180885 Added support to create asymmetrical U-Net structures (#5400) ...

* Added args, kwargs to ```U

* Add UNetMidBlock2D as a supported mid block type

* Fix extra init input for UNetMidBlock2D, change allowed types for Mid-block init

* Update unet_2d_condition.py

* Update unet_2d_condition.py

* Update unet_2d_condition.py

* Update unet_2d_condition.py

* Update unet_2d_condition.py

* Update unet_2d_condition.py

* Update unet_2d_condition.py

* Update unet_2d_condition.py

* Update unet_2d_blocks.py

* Update unet_2d_blocks.py

* Update unet_2d_blocks.py

* Update unet_2d_condition.py

* Update unet_2d_blocks.py

* Updated docstring, increased check strictness

Updated the docstring for ```UNet2DConditionModel``` to include ```reverse_transformer_layers_per_block``` and updated checking for nested list type ```transformer_layers_per_block```

* Add basic shape-check test for asymmetrical unets

* Update src/diffusers/models/unet_2d_blocks.py

Removed blank line

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

* Update unet_2d_condition.py

Remove blank space

* Update unet_2d_condition.py

Changed docstring for `mid_block_type`

* Fixed docstring and wrong default value

* Reformat with black

* Reformat with necessary commands

* Add UNetMidBlockFlat to versatile_diffusion/modeling_text_unet.py to ensure consistency

* Removed args, kwargs, use on mid-block type

* Make fix-copies

* Update src/diffusers/models/unet_2d_condition.py

Wrap into single line

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

* make fix-copies

---------

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

2023-10-20 11:42:28 +02: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_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

New xformers test runner (#5349)

2023-10-13 00:32:39 +05:30

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

Added support to create asymmetrical U-Net structures (#5400)

2023-10-20 11:42:28 +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

[LoRA] Centralize LoRA tests (#5086)

2023-09-18 17:54:33 +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