Skip to content

Commit fbf82b0

Browse files
Update docs/source/trl_integration.md
Co-authored-by: Quentin Gallouédec <45557362+qgallouedec@users.noreply.github.com>
1 parent f40031c commit fbf82b0

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

docs/source/trl_integration.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@ from datasets import Dataset
77
from trl import SFTConfig, SFTTrainer
88

99
# Create a small fake dataset
10-
prompts = ["What is the capital of France?", "Who wrote Hamlet?"] * 12
11-
completions = ["Paris.", "Shakespeare."] * 12
10+
prompts = ["The capital of France is", "Hamlet was written by"] * 12
11+
completions = [" Paris.", " Shakespeare."] * 12
1212
dataset = Dataset.from_dict({"prompt": prompts, "completion": completions})
1313

1414
# Train a model using the TRL SFTTrainer API

0 commit comments

Comments
 (0)