Merge branch 'main' into chcui/fix_mcore_conversion_bug

NVIDIA · Nov 2, 2023 · c6b4e54 · c6b4e54
2 parents d8c7192 + 05ecfe4
commit c6b4e54
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/nemo/utils/model_utils.py b/nemo/utils/model_utils.py
@@ -624,6 +624,10 @@ def ckpt_to_dir(filepath: Union[str, Path]) -> Path:
 
     filepath = Path(filepath)
 
+    # if it is already a distributed checkpoint, then return
+    if filepath.suffix != ".ckpt" and filepath.is_dir():
+        return filepath
+
     # adding this assert because we will later remove directories based on the return value of this method
     assert filepath.suffix == ".ckpt", f'filepath: {filepath} must have .ckpt extension'