Skip to content

Commit 06f28ea

Browse files
fixup! fixup! fix: only trigger on_save if the output checkpoint path actually contains files
Signed-off-by: Harikrishnan Balagopal <harikrishmenon@gmail.com>
1 parent ae34e7a commit 06f28ea

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tuning/sft_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -587,7 +587,7 @@ def save(path: str, trainer: SFTTrainer, tc_callback, log_level="WARNING", args=
587587
path,
588588
)
589589
actually_saved = len(saved_files) > 0
590-
except Exception as e:
590+
except Exception as e: # pylint: disable-next=broad-exception-caught
591591
logger.warning(
592592
"sanity check, failed to list files in checkpoint path '%s' , error: %s",
593593
path,

0 commit comments

Comments
 (0)