Skip to content

Commit

Permalink
fix padding in dpo trainer (#1284)
Browse files Browse the repository at this point in the history
  • Loading branch information
pacman100 authored Jan 30, 2024
1 parent aa35fec commit 9186710
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion trl/trainer/dpo_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -146,7 +146,7 @@ def __init__(
args: TrainingArguments = None,
data_collator: Optional[DataCollator] = None,
label_pad_token_id: int = -100,
padding_value: int = 0,
padding_value: Optional[int] = None,
truncation_mode: str = "keep_end",
train_dataset: Optional[Dataset] = None,
eval_dataset: Optional[Union[Dataset, Dict[str, Dataset]]] = None,
Expand Down

0 comments on commit 9186710

Please sign in to comment.