fix bug

Former-commit-id: 3164b4f11b
2025-12-14 19:06:26 +08:00 · 2024-03-26 17:30:12 +08:00
parent ec94e5e876
commit c311375b50
1 changed files with 1 additions and 1 deletions
--- a/src/llmtuner/hparams/finetuning_args.py
+++ b/src/llmtuner/hparams/finetuning_args.py
@@ -102,7 +102,7 @@ class RLHFArguments:
        default="sigmoid",
        metadata={"help": "The type of DPO loss to use."},
    )
-    dpo_label_smoothing = field(
+    dpo_label_smoothing: float = field(
        default=0.0,
        metadata={"help": "The robust DPO label smoothing parameter in cDPO that should be between 0 and 0.5."},
    )