src/diffusers/pipelines/pipeline_utils.py · e2deb82e6925a861c9414894542b20251a37fc99 · renzhc / diffusers_dcu

"vscode:/vscode.git/clone" did not exist on "86fe4fa756d9e46d6fa1102bcca46836093f72b8"

Fix compatibility with pipeline when loading model with device_map on single gpu (#10390) · e2deb82e

Marc Sun authored Jan 08, 2025



* fix device issue in single gpu case

* Update src/diffusers/pipelines/pipeline_utils.py
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

---------
Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

e2deb82e

pipeline_utils.py 94.2 KB

Replace pipeline_utils.py