fix bug
This commit is contained in:
1 parent
511f675402
commit
3164b4f11b
1 file changed
+1
-1
@@ -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."},
|
||||
)
|
||||
|
||||
Reference in new issue
Block a user