Aryan 71ad16b463 Add _no_split_modules to some models (#10308) ...

* set supports gradient checkpointing to true where necessary; add missing no split modules

* fix cogvideox tests

* update

---------

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

2025-01-08 06:34:19 +05:30

..

autoencoders

[core] LTX Video 0.9.1 (#10330)

2024-12-23 19:51:33 +05:30

transformers

Add _no_split_modules to some models (#10308)

2025-01-08 06:34:19 +05:30

unets

Enable Gradient Checkpointing for UNet2DModel (New) (#7201)

2024-12-19 14:45:45 -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

[Tests] Fix more tests sayak (#10359)

2024-12-23 19:11:21 +05:30

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

[Sana bug] bug fix for 2K model config (#10340)

2024-12-23 08:56:25 +05:30