Skip to content

Commit 1f158f8

Browse files
Change checkpoint naming check in example script
Co-authored-by: Oleg Silkin <[email protected]>
1 parent 58dadcf commit 1f158f8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/scripts/osft_gpt_oss_example.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def find_most_recent_checkpoint(output_dir):
3939
ValueError: If no checkpoints are found
4040
"""
4141
# Get all checkpoint directories under hf_format
42-
checkpoint_pattern = os.path.join(output_dir, "hf_format", "samples_*.0")
42+
checkpoint_pattern = os.path.join(output_dir, "hf_format", "samples_*")
4343
checkpoint_dirs = glob.glob(checkpoint_pattern)
4444

4545
if not checkpoint_dirs:

0 commit comments

Comments
 (0)