Skip to content

Commit 838f280

Browse files
authored
Fix sft post-training run name can not be identified (#1156)
Fix sft post-training run name so it can be identified and upload to vertex.
1 parent b9c513a commit 838f280

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

dags/post_training/maxtext_sft.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -152,7 +152,7 @@ def validate_training(
152152

153153
for slice_num in training_config.slices:
154154
current_datetime = datetime.datetime.now().strftime("%Y-%m-%d-%H-%M-%S")
155-
run_name = f"{mode.value}-{current_datetime}"
155+
run_name = f"sft-{mode.value}-{current_datetime}"
156156

157157
sft_training_command = training_config.generate_sft_training_command(
158158
run_name=run_name,

0 commit comments

Comments
 (0)