fix bug

hiyouga · Mar 26, 2024 · 3164b4f · 3164b4f
1 parent 511f675
commit 3164b4f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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."},
     )