Skip to content

Commit 4dcb2d4

Browse files
firefighter-ericJintao-Huang
authored andcommitted
fix: fix trainer state strict saving (#4995)
1 parent 1dda899 commit 4dcb2d4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/llm/train/sft.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -173,7 +173,7 @@ def _save_trainer_state(self, trainer):
173173
})
174174
if is_master():
175175
jsonl_path = os.path.join(training_args.output_dir, 'logging.jsonl')
176-
append_to_jsonl(jsonl_path, self.train_msg)
176+
append_to_jsonl(jsonl_path, self.train_msg, strict=False)
177177
return self.train_msg
178178

179179
def train(self, trainer):

0 commit comments

Comments
 (0)