quickstart.py

from trl import SFTTrainer
from datasets import load_dataset

dataset = load_dataset("trl-lib/Capybara", split="train")

trainer = SFTTrainer(
    model="openai-community/gpt2",
    train_dataset=dataset,
)
trainer.train()