Sayak Paul 3fd31eef51 [Core] introduce _no_split_modules to ModelMixin (#6396) ...

* introduce _no_split_modules.

* unnecessary spaces.

* remove unnecessary kwargs and style

* fix: accelerate imports.

* change to _determine_device_map

* add the blocks that have residual connections.

* add: CrossAttnUpBlock2D

* add: testin

* style

* line-spaces

* quality

* add disk offload test without safetensors.

* checking disk offloading percentages.

* change model split

* add: utility for checking multi-gpu requirement.

* model parallelism test

* splits.

* splits.

* splits

* splits.

* splits.

* splits.

* offload folder to test_disk_offload_with_safetensors

* add _no_split_modules

* fix-copies

2024-04-30 08:46:51 +05:30

..

autoencoders

Fix test for consistency decoder. (#7746)

2024-04-24 12:28:11 +05:30

transformers

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

unets

[Core] introduce _no_split_modules to ModelMixin (#6396)

2024-04-30 08:46:51 +05:30

__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

[LoRA] use the PyTorch classes wherever needed and start depcrecation cycles (#7204)

2024-03-13 07:56:19 +05:30

test_modeling_common_flax.py

[Tests] parallelize (#3078)

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

test_modeling_common.py

[Core] introduce _no_split_modules to ModelMixin (#6396)

2024-04-30 08:46:51 +05:30