transformers

9edf3758 - [GIT] Fix training (#21133)

[GIT] Fix training (#21133) * Fix training * Add test * Fix failing tests Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>