transformers
Create the return value on device to avoid unnecessary copying from CPU
#26151

Merged

Create the return value on device to avoid unnecessary copying from CPU #26151

ArthurZucker merged 1 commit into huggingface:main from perf_fix

Create the return value on device to avoid unnecessary copying from CPU

9eda98d9

ArthurZucker commented on 2023-09-13

ArthurZucker requested a review from

fxmarty 2 years ago

fxmarty approved these changes on 2023-09-16

fxmarty requested a review from

ArthurZucker 2 years ago

ArthurZucker approved these changes on 2023-09-18

ArthurZucker merged 97f439ae into main 2 years ago

Reviewers

ArthurZucker

fxmarty

Assignees

No one assigned

Labels

None yet

Milestone

No milestone