transformers
Fix Unnecessary move of tensors from CPU to GPU in LlamaRotaryEmbedding
#22234

Merged

Fix Unnecessary move of tensors from CPU to GPU in LlamaRotaryEmbedding #22234

sgugger merged 1 commit into huggingface:main from main

push

239da8fb

younesbelkada approved these changes on 2023-03-17

younesbelkada requested a review from

amyeroberts 3 years ago

amyeroberts approved these changes on 2023-03-17

sgugger approved these changes on 2023-03-17

sgugger merged cf601b90 into main 3 years ago

BlackSamorez commented on 2023-03-21

Reviewers

younesbelkada

sgugger

amyeroberts

BlackSamorez

Assignees

No one assigned

Labels

None yet

Milestone

No milestone