update smddp api to v1.4.0 (#16371)
* update smddp api to v1.4.0
* Update src/transformers/trainer.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* Update src/transformers/trainer.py
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
* address comments
* fix style
* remove unused import
* fix indent
* disable style check for import
* fix space
Co-authored-by: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>