fix deepspeed load best model at end when the model gets sharded
# What does this PR do?
1. Fixes https://github.com/huggingface/transformers/issues/25027
_The documentation is not available anymore as the PR was closed or merged._