Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
qgallouedec committed May 31, 2024
1 parent e1b8755 commit 8075419
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion trl/trainer/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -312,7 +312,7 @@ def pad(tensors: List[torch.Tensor], padding_value: int = 0, padding_side: str =
Returns:
`torch.Tensor`:
A single tensor containing the padded tensors.
Examples:
>>> import torch
>>> pad([torch.tensor([1, 2, 3]), torch.tensor([4, 5])])
Expand Down

0 comments on commit 8075419

Please sign in to comment.