PR #28326 Add the XPU device check for pipeline mode

Add the XPU device check for pipeline mode #28326

amyeroberts merged 7 commits into huggingface:main from yuanwu2017:xpu

ArthurZucker requested a review from

ArthurZucker 2 years ago

ArthurZucker approved these changes on 2024-01-09

yuanwu2017 commented on 2024-01-10

yuanwu2017 force pushed 2 years ago

ArthurZucker commented on 2024-01-11

yuanwu2017 force pushed to 83a00b3d 2 years ago

amyeroberts merged a573ac74 into main 2 years ago

Add the XPU check for pipeline mode

e5079304

Don't move model to device when hf_device_map isn't None

4e738524

Raise the error when xpu is not available

688ba52c

Update src/transformers/pipelines/base.py

3145caf1

Update src/transformers/pipelines/base.py

2da5cbc3

Modify the error message

0924141f

Change message format.

83a00b3d

Reviewers

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone