From a916688723cd9d39c0efe34d51e1e7bfd9b4024c 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: f513e1415cc3fe87f600318fba855d1286b6d007 --- 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."}, )