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

Merge branch 'main' into feature/full-finetuning

Browse Source
This commit is contained in:
Sayak Paul
2025-01-10 19:56:34 +05:30
committed by GitHub
parent 34da4c5a6b b34091362e
commit 014960fd4c
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 # before we save the new checkpoint, we need to have at_most `checkpoints_total_limit - 1` checkpoints
if len(checkpoints) >= checkpointing_limit: if len(checkpoints) >= checkpointing_limit:
num_to_remove = len(checkpoints) - checkpointing_limit + 1 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) delete_files(checkpoints_to_remove)
logger.info(f"Checkpointing at step {step}") logger.info(f"Checkpointing at step {step}")