tests/pipelines/test_pipelines.py · 30a512ea699f307d5c0cacc4858d97fc0dd2afb6 · renzhc / diffusers_dcu

[Core] Improve `.to(...)` method, fix offloads multi-gpu, add docstring, add dtype (#5132) · 30a512ea

Patrick von Platen authored Sep 25, 2023



* fix cpu offload

* fix

* fix

* Update src/diffusers/pipelines/pipeline_utils.py

* make style

* Apply suggestions from code review
Co-authored-by: YiYi Xu <yixu310@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* fix more

* fix more

---------
Co-authored-by: YiYi Xu <yixu310@gmail.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

30a512ea

test_pipelines.py 77.2 KB

Replace test_pipelines.py