diffusers

mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00

Files

Pi Esposito 4d1cce2fd0 add accelerate to load models with smaller memory footprint (#361 )

* add accelerate to load models with smaller memory footprint

* remove low_cpu_mem_usage as it is reduntant

* move accelerate init weights context to modelling utils

* add test to ensure results are the same when loading with accelerate

* add tests to ensure ram usage gets lower when using accelerate

* move accelerate logic to single snippet under modelling utils and remove it from configuration utils

* format code using to pass quality check

* fix imports with isor

* add accelerate to test extra deps

* only import accelerate if device_map is set to auto

* move accelerate availability check to diffusers import utils

* format code

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2022-10-04 15:21:40 +02:00

__init__.py

fix issues with loading, add test for pipeline

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

conftest.py

[Utils] Add deprecate function and move testing_utils under utils (#659 )

2022-10-03 23:44:24 +02:00

test_config.py

[Tests] Make sure tests are on GPU (#269 )

2022-08-29 15:58:11 +02:00

test_layers_utils.py

[Utils] Add deprecate function and move testing_utils under utils (#659 )