PR #47503 Fix TP inference for tied embedding

Fix TP inference for tied embedding #47503

3outeille merged 13 commits into main from fix/tp-colwise-gather-output

fix tied embedding colwise

04ae0430

merge conflict

6c740ce3

linting

1d6731fb

3outeille marked this pull request as ready for review 33 days ago

Merge branch 'main' into fix/tp-colwise-gather-output

63164554

github-actions requested a review from

ArthurZucker 33 days ago

github-actions requested a review from

Rocketknight1 33 days ago

3outeille removed review request from

Rocketknight1 33 days ago

vasqu commented on 2026-07-23

Merge branch 'main' into fix/tp-colwise-gather-output

1b368962

edit in super class

008e5743

edit super class

b95cf415

revert

ea1bd11c

revert

3bd0221b

Merge branch 'main' of https://github.com/huggingface/transformers in…

eb4aed3e

revert

bbdc659b

revert to update main not local

28c0628c

3outeille changed the title ~~Fix inference for tied embedding~~ Fix TP inference for tied embedding 32 days ago

vasqu approved these changes on 2026-07-24

Merge branch 'main' into fix/tp-colwise-gather-output

fb147eca

3outeille enabled auto-merge 29 days ago

3outeille merged 279e6efc into main 29 days ago

3outeille deleted the fix/tp-colwise-gather-output branch 29 days ago

Reviewers

vasqu

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

transformers Fix TP inference for tied embedding #47503 Merged

Fix TP inference for tied embedding #47503

transformers
Fix TP inference for tied embedding
#47503

Merged