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

Commit
345 days ago
Fix compatibility with pipeline when loading model with device_map on single gpu (#10390) * 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>
Author
Parents
Loading