Skip to content

Commit 845617a

Browse files
authored
add backward compatibility support for loading mcore 0.15 checkpoints (NVIDIA#2648)
Signed-off-by: dimapihtar <[email protected]>
1 parent 0a59bea commit 845617a

File tree

1 file changed

+7
-0
lines changed
  • megatron/core/dist_checkpointing/strategies

1 file changed

+7
-0
lines changed

megatron/core/dist_checkpointing/strategies/torch.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,13 @@
8383
_metadata_fn: str = ".metadata"
8484

8585

86+
# dummy class needed to load mcore v0.15 checkpoints with optim. states
87+
class MCoreMetadata:
88+
""" """
89+
90+
pass
91+
92+
8693
# dummy class needed to load old checkpoint weights
8794
class MCoreSavePlan:
8895
""" """

0 commit comments

Comments
 (0)