transformers
multi-gpu: fix tensor device placements for various models
#35763

Merged

multi-gpu: fix tensor device placements for various models #35763

ArthurZucker merged 3 commits into huggingface:main from dvrogozh:multi-gpu2

dvrogozh requested a review from

zucchini-nlp 362 days ago

dvrogozh requested a review from

ArthurZucker 362 days ago

dvrogozh requested a review from

amyeroberts 362 days ago

dvrogozh requested a review from

qubvel 362 days ago

dvrogozh commented on 2025-01-18

dvrogozh requested a review from

eustlb 362 days ago

dvrogozh requested a review from

Cyrilvallez 362 days ago

dvrogozh requested a review from

Rocketknight1 362 days ago

SunMarc commented on 2025-01-20

qubvel removed review request from

qubvel 359 days ago

dvrogozh force pushed from 8ce6e2f2 to eae83792 357 days ago

SunMarc approved these changes on 2025-02-04

milti-gpu: fix inputs_embeds + position_embeds

dcdb6013

multi-gpu: fix tensor device placements for various models

2c438859

dvrogozh force pushed from eae83792 to b3541dd8 344 days ago

Apply make fix-copies

debb2e47

dvrogozh force pushed from b3541dd8 to debb2e47 344 days ago

ArthurZucker approved these changes on 2025-02-12

ArthurZucker merged 24405127 into main 336 days ago

ArthurZucker removed review request from

zucchini-nlp 336 days ago

ArthurZucker removed review request from

amyeroberts 336 days ago

ArthurZucker removed review request from

eustlb 336 days ago

ArthurZucker removed review request from

Cyrilvallez 336 days ago

ArthurZucker removed review request from

Rocketknight1 336 days ago

Reviewers

ArthurZucker

SunMarc

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers multi-gpu: fix tensor device placements for various models #35763 Merged

multi-gpu: fix tensor device placements for various models #35763

transformers
multi-gpu: fix tensor device placements for various models
#35763

Merged