PR #47253 🚨 Pass tp_plan from lm_heads

🚨 Pass tp_plan from lm_heads #47253

ArthurZucker merged 5 commits into main from fix_tp_plans_for_lm_heads

pass tp_plan from lm_heads

273c2e1e

add a small test to avoid regressiosn on TP

08ab9ab3

molbap requested a review from

3outeille 7 days ago

molbap requested a review from

ArthurZucker 7 days ago

world size must divide vocab size

3c326e4a

molbap commented on 2026-07-10

molbap changed the title ~~Pass tp_plan from lm_heads~~ 🚨 Pass tp_plan from lm_heads 7 days ago

early raise for incompatible vocabs

ae6881e1

test our test

d51432ad

vasqu approved these changes on 2026-07-10

3outeille approved these changes on 2026-07-13

ArthurZucker commented on 2026-07-13

ArthurZucker merged f5a67a46 into main 2 days ago

ArthurZucker deleted the fix_tp_plans_for_lm_heads branch 2 days ago

Reviewers

3outeille

vasqu

ArthurZucker

Assignees

No one assigned

Labels

None yet

Milestone

No milestone