Skip to content

Commit

Permalink
Update composer/utils/checkpoint.py
Browse files Browse the repository at this point in the history
Co-authored-by: Mihir Patel <[email protected]>
  • Loading branch information
j316chuck and mvpatel2000 authored Oct 14, 2024
1 parent b23b2a0 commit 95624dd
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions composer/utils/checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -1015,11 +1015,11 @@ def safe_torch_load(
raise e
except FileNotFoundError as e:
if 'No such file or directory' in str(e) and dist.get_local_rank() != 0:
global_rank_zeor = dist.get_global_rank() - dist.get_local_rank()
global_rank_zero = dist.get_global_rank() - dist.get_local_rank()
raise FileNotFoundError(
f'No such file or directory: {e.filename}. '
f'This likely implies a download failed on local rank 0 or global rank {global_rank_zeor}'
f'Please check the logs for global rank {global_rank_zeor} to debug the checkpoint download issue.',
f'This likely implies a download failed on local rank 0 or global rank {global_rank_zero}'
f'Please check the logs for global rank {global_rank_zero} to debug the checkpoint download issue.',
) from e
raise e

Expand Down

0 comments on commit 95624dd

Please sign in to comment.