Fix push_to_hub for TPUs
# What does this PR do?

This PR fixes the `push_to_hub` method for TPUs, which hangs forever right now because there is a reandezvous point in code that is only reached by the main process.