Sayak Paul c1926cef6b [tests] refactor vae tests (#9808) ...

* add: autoencoderkl tests

* autoencodertiny.

* fix

* asymmetric autoencoder.

* more

* integration tests for stable audio decoder.

* consistency decoder vae tests

* remove grad check from consistency decoder.

* cog

* bye test_models_vae.py

* fix

* fix

* remove allegro

* fixes

* fixes

* fixes

---------

Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>

2024-12-04 15:58:36 +05:30

..

autoencoders

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30

transformers

add skip_layers argument to SD3 transformer model class (#9880)

2024-11-19 15:22:54 -05:00

unets

[LoRA] feat: save_lora_adapter() (#9862)

2024-11-18 21:03:38 -10: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

adding back test_conversion_when_using_device_map (#7704)

2024-04-18 19:21:32 -10:00

test_layers_utils.py

Support bfloat16 for Upsample2D (#9480)

2024-10-01 16:08:12 -10:00

test_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

[tests] refactor vae tests (#9808)

2024-12-04 15:58:36 +05:30