Don't download if we don't have to

allenai · Oct 8, 2023 · c110059 · c110059
1 parent d22ef89
commit c110059
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/olmo/train.py b/olmo/train.py
@@ -672,7 +672,9 @@ def restore_checkpoint(
             )
         elif checkpoint_type == CheckpointType.sharded or checkpoint_type is None:
             try:
-                legacy_mode = resource_path(load_path, f"rank{get_global_rank()}.pt").is_file()
+                legacy_mode = resource_path(
+                    load_path, f"rank{get_global_rank()}.pt", local_cache=local_cache
+                ).is_file()
             except FileNotFoundError:
                 legacy_mode = False
             if legacy_mode: