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

..

fixtures

change to 2024 in the license (#6902)

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

lora

FIX Setting device for DoRA parameters (#7655)

2024-04-12 13:55:46 +02:00

models

[Core] introduce _no_split_modules to ModelMixin (#6396)

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

others

Remove distutils (#7455)

2024-03-25 06:44:53 +05:30

pipelines

[Tests] reduce the model size in the amused fast test (#7804)

2024-04-30 08:11:26 +05:30

schedulers

UniPC Multistep add rescale_betas_zero_snr (#7531)

2024-04-02 17:23:55 -10:00

__init__.py

fix issues with loading, add test for pipeline

2022-06-07 15:40:36 +02:00

conftest.py

change to 2024 in the license (#6902)

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