diffusers
Fix compatibility with pipeline when loading model with device_map on single gpu
#10390

Merged

Fix compatibility with pipeline when loading model with device_map on single gpu #10390

SunMarc merged 2 commits into main from remove-force-hook-from-model-init

fix device issue in single gpu case

0dbca3fa

SunMarc commented on 2024-12-26

SunMarc requested a review from

sayakpaul 1 year ago

SunMarc requested a review from

yiyixuxu 1 year ago

sayakpaul commented on 2024-12-26

Update src/diffusers/pipelines/pipeline_utils.py

54a903be

yiyixuxu approved these changes on 2025-01-08

SunMarc merged e2deb82e into main 1 year ago

sayakpaul deleted the remove-force-hook-from-model-init branch 1 year ago

Reviewers

yiyixuxu

sayakpaul

Assignees

No one assigned

Labels

None yet

Milestone

No milestone