transformers

f5a67a46 - 🚨 Pass tp_plan from lm_heads (#47253)

🚨 Pass tp_plan from lm_heads (#47253) * pass tp_plan from lm_heads * add a small test to avoid regressiosn on TP * world size must divide vocab size * early raise for incompatible vocabs * test our test