fix mutable field issue

huggingface · Jan 23, 2025 · 7638607 · 7638607
1 parent c7a5511
commit 7638607
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/trl/trainer/dpo_config.py b/trl/trainer/dpo_config.py
@@ -287,6 +287,9 @@ class DPOConfig(TrainingArguments):
     loss_weights: Optional[Dict[str, float]] = field(
         default={"sigmoid": 1.0},
     )
+    loss_weights: Optional[Dict[str, float]] = field(
+        default_factory=lambda: ["your_values"]
+    )
     loss_type: List[str] | str = field(
         default="sigmoid",
         metadata={