From c311375b507827b25803ddd5db6990784da1d2c2 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Tue, 26 Mar 2024 17:30:12 +0800 Subject: [PATCH] fix bug Former-commit-id: 3164b4f11b72684c8aa2105037cb36c47b6acfd4 --- src/llmtuner/hparams/finetuning_args.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/llmtuner/hparams/finetuning_args.py b/src/llmtuner/hparams/finetuning_args.py index 6fe951f1..0dd28a8f 100644 --- 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."}, )