transformers

20ce210a - Revert "remove dtensors, not explicit (#39840)" (#39912)

Revert "remove dtensors, not explicit (#39840)" (#39912) * Revert "remove dtensors, not explicit (#39840)" This did not work with generation (lm_head needs extra care!) This reverts commit 6dfd561d9cd722dfc09f702355518c6d09b9b4e3. * update * style?