Fix checkpoint restoring
# What does this PR do ?

Fix restoring issue with PTL 1.7  

**Collection**: Core

# Changelog 
Move all non-zero rank value checkpoints to rank 0 checkpoints before restoring. 


**PR Type**:
- [ ] New Feature
- [x] Bugfix
- [ ] Documentation 