Fix default value for training_args in FinetuningConfig

T-bagwell · Mar 20, 2024 · 46e3ed3 · 46e3ed3
1 parent c814a72
commit 46e3ed3
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