Backwards compatibility

allenai · Dec 16, 2023 · 933cce4 · 933cce4
1 parent 4935c0f
commit 933cce4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/olmo/train.py b/olmo/train.py
@@ -166,7 +166,7 @@ def load_trainer_state_dict(self, state_dict: Dict[str, Any]) -> None:
         ]
         self.ephemeral_checkpoints = [
             path
-            for path in state_dict["ephemeral_checkpoints"]
+            for path in state_dict.get("ephemeral_checkpoints", [])
             if path.is_dir() and path.resolve().parent == Path(self.cfg.save_folder).resolve()
         ]