diffusers/tests
Suraj Patil 7482178162
default fast model loading 🔥 (#1115)
* make accelerate hard dep

* default fast init

* move params to cpu when device map is None

* handle device_map=None

* handle torch < 1.9

* remove device_map="auto"

* style

* add accelerate in torch extra

* remove accelerate from extras["test"]

* raise an error if torch is available but not accelerate

* update installation docs

* Apply suggestions from code review

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

* improve defautl loading speed even further, allow disabling fats loading

* address review comments

* adapt the tests

* fix test_stable_diffusion_fast_load

* fix test_read_init

* temp fix for dummy checks

* Trigger Build

* Apply suggestions from code review

Co-authored-by: Anton Lozhkov <anton@huggingface.co>

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
Co-authored-by: Anton Lozhkov <anton@huggingface.co>
2022-11-03 17:25:57 +01:00
..
fixtures/custom_pipeline Custome Pipelines (#744) 2022-10-06 16:54:02 +02:00
models default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
pipelines default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
repo_utils default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01: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 Fix hub-dependent tests for PRs (#1119) 2022-11-03 15:24:32 +01:00
test_layers_utils.py VQ-diffusion (#658) 2022-11-03 16:10:28 +01:00
test_modeling_common.py Add Apple M1 tests (#796) 2022-10-17 20:27:30 +02:00
test_modeling_common_flax.py Flax: Trickle down `norm_num_groups` (#789) 2022-10-11 20:05:10 +02:00
test_outputs.py Fix BaseOutput initialization from dict (#570) 2022-09-20 18:32:16 +02:00
test_pipelines.py default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
test_pipelines_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_pipelines_flax.py [Loading] Ignore unneeded files (#1107) 2022-11-02 19:20:42 +01:00
test_pipelines_onnx_common.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00
test_scheduler.py VQ-diffusion (#658) 2022-11-03 16:10:28 +01:00
test_scheduler_flax.py [CI] Framework and hardware-specific CI tests (#997) 2022-11-02 14:07:07 +01:00
test_training.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00
test_utils.py [Utils] Add deprecate function and move testing_utils under utils (#659) 2022-10-03 23:44:24 +02:00