Allows non-strict load with distributed checkpoints#9715

mikolajblaz merged 1 commit intomainNVIDIA-NeMo/NeMo:mainfrom

cherry-pick-main-423951179e73fb29f77e28120e38708535098152NVIDIA-NeMo/NeMo:cherry-pick-main-423951179e73fb29f77e28120e38708535098152Copy head branch name to clipboard