transformers
Warn on TPUs when the custom optimizer and model device are not the same
#18668
Merged
Merged