Merge pull request THUDM#1008 from chatgpt-1/main

Fix default value for training_args in FinetuningConfig
T-bagwell · Mar 21, 2024 · 018267b · 018267b
2 parents c814a72 + 46e3ed3
commit 018267b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/finetune_demo/finetune_hf.py b/finetune_demo/finetune_hf.py
@@ -151,7 +151,7 @@ class FinetuningConfig(object):
     max_output_length: int
 
     training_args: Seq2SeqTrainingArguments = dc.field(
-        default=Seq2SeqTrainingArguments(output_dir='./output')
+        default_factory=lambda: Seq2SeqTrainingArguments(output_dir='./output')
     )
     peft_config: Optional[PeftConfig] = None