diffusers/tests/pipelines
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
..
dance_diffusion default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
ddim default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
ddpm default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
karras_ve default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
latent_diffusion default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
pndm default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
repaint feat: add repaint (#974) 2022-11-03 15:42:46 +01:00
score_sde_ve default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
stable_diffusion default fast model loading 🔥 (#1115) 2022-11-03 17:25:57 +01:00
vq_diffusion VQ-diffusion (#658) 2022-11-03 16:10:28 +01:00
__init__.py Reorganize pipeline tests (#963) 2022-10-24 16:34:01 +02:00