From e145a2ce0ca3b2aec403ff16c8bff6d812b48c33 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Thu, 21 Sep 2023 19:52:06 +0800 Subject: [PATCH] tiny fix Former-commit-id: d24ea58c1a44b94227f4cb60f13fc1dd79997d01 --- src/llmtuner/tuner/sft/workflow.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/tuner/sft/workflow.py b/src/llmtuner/tuner/sft/workflow.py index 05942780..d45571d2 100644 --- a/src/llmtuner/tuner/sft/workflow.py +++ b/src/llmtuner/tuner/sft/workflow.py @@ -39,7 +39,7 @@ def run_sft( # Override the decoding parameters of Seq2SeqTrainer training_args_dict = training_args.to_dict() training_args_dict.update(dict( - generation_max_length=training_args.generation_max_length or data_args.max_target_length, + generation_max_length=training_args.generation_max_length or data_args.cutoff_len, generation_num_beams=data_args.eval_num_beams or training_args.generation_num_beams )) training_args = Seq2SeqTrainingArguments(**training_args_dict)