storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-09 00:09:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix the checkpoint dir bug in get_intermediate_ckpt_path so that checkpointing_limit can actually work. (#207)

Browse Source
This commit is contained in:
Boyang Deng
2025-01-09 23:48:32 -08:00
committed by GitHub
parent 4b21d37c9d
commit b34091362e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/utils/checkpointing.py
+1 -1
View File
@@ -55,7 +55,7 @@ def get_intermediate_ckpt_path(checkpointing_limit: int, step: int, output_dir:
# before we save the new checkpoint, we need to have at_most `checkpoints_total_limit - 1` checkpoints
if len(checkpoints) >= checkpointing_limit:
num_to_remove = len(checkpoints) - checkpointing_limit + 1
checkpoints_to_remove = checkpoints[0:num_to_remove]
checkpoints_to_remove = [os.path.join(output_dir, x) for x in checkpoints[0:num_to_remove]]
delete_files(checkpoints_to_remove)
logger.info(f"Checkpointing at step {step}")