Skip to content

Commit

Permalink
Fix weird doc bug (#1244)
Browse files Browse the repository at this point in the history
* Update utils.py

* Update trl/trainer/utils.py

* Update trl/trainer/utils.py
  • Loading branch information
younesbelkada authored Jan 18, 2024
1 parent 4fb3d0c commit 3319993
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 @@ -367,7 +367,7 @@ class ConstantLengthDataset(IterableDataset):
Name of the field in the dataset that contains the text. Used only if `formatting_func` is `None`.
formatting_func (`Callable`, **optional**):
Function that formats the text before tokenization. Usually it is recommended to have follows a certain
pattern such as `"### Question: {question}\n ### Answer: {answer}\n"`
pattern such as `"### Question: {question} ### Answer: {answer}"`
infinite (`bool`, *optional*, defaults to `False`):
If True the iterator is reset after dataset reaches end else stops.
seq_length (`int`, *optional*, defaults to `1024`):
Expand Down

0 comments on commit 3319993

Please sign in to comment.