ready

TJ-Solergibert · Nov 26, 2024 · bd81b67 · bd81b67
1 parent d5f656a
commit bd81b67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nanotron/serialize/main.py b/src/nanotron/serialize/main.py
@@ -203,7 +203,7 @@ def save(
     dist.barrier(parallel_context.world_pg)
 
 
-ckpt_path(config: Config, parallel_context: ParallelContext) -> Optional[Path]:
+def parse_ckpt_path(config: Config, parallel_context: ParallelContext) -> Optional[Path]:
     """Parse checkpoint path from config and download checkpoint from S3 if needed.
 
     Args: