find_checkpoint_rank_0 returns a single value

RaymondLi0 · RaymondLi0 · commit 0229a695488f · 2023-06-19T09:45:16.000-04:00
diff --git a/megatron/checkpointing.py b/megatron/checkpointing.py
@@ -140,7 +140,7 @@ def find_checkpoint_rank_0(checkpoints_path, iteration, release=False):
     if os.path.isfile(filename):
         return filename
 
-    return None, None
+    return None
 
 
 def get_checkpoint_tracker_filename(checkpoints_path):
