src/diffusers/pipelines/pipeline_utils.py · 3279751bf946b283f739c03f6248f169ce57ab8f · renzhc / diffusers_dcu

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