From 3bdc40b4db50693502237805c451cae4a32fb765 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: dbaef776a171dfb316567553788f2d81d35a51fb --- 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)