accelerate
(Part 1) fix: make TP training compatible with new transformers
#3457
Merged
Merged