src/diffusers/models/modeling_utils.py · a6f043a80f4951bb65ddb05769723fddb0303a9b · 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

modeling_utils.py 68.8 KB

Replace modeling_utils.py