accelerate
only move model to device when model is in cpu and target device is xpu
#3133
Merged

only move model to device when model is in cpu and target device is xpu #3133

muellerzr merged 1 commit into huggingface:main from faaany:ipex-bug
faaany
faaany bug fix
2471af35
faaany
faaany faaany changed the title fix tensor device misalignment on xpu for model loaded with `device_map="auto"` only move model to device when model is in cpu and target device in xpu 1 year ago
faaany faaany changed the title only move model to device when model is in cpu and target device in xpu only move model to device when model is in cpu and target device is xpu 1 year ago
yao-matrix
SunMarc
SunMarc approved these changes on 2024-09-30
HuggingFaceDocBuilderDev
muellerzr
muellerzr approved these changes on 2024-10-07
muellerzr muellerzr merged 10776115 into main 1 year ago
faaany faaany deleted the ipex-bug branch 1 year ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone