Final singlefile checkpoint saves one folder up

Signed-off-by: Davis Wertheimer <[email protected]>
foundation-model-stack · Jan 14, 2025 · 292b459 · 292b459
1 parent 0e47e93
commit 292b459
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fms_fsdp/utils/checkpointing_utils.py b/fms_fsdp/utils/checkpointing_utils.py
@@ -324,7 +324,7 @@ def save_single_file(
     ):
         # Note: metadata kwargs cannot contain any of:
         # (step, model)
-        save_name = os.path.join(self.ckp_path, "step_" + str(step) + "_ckp.pth")
+        save_name = os.path.join(self.ckp_path[:-12], "step_" + str(step) + "_ckp.pth")
         save_time = time.time()
         with FSDP.state_dict_type(
             model,