Skip to content

tinker_cookbook.supervised

Class Description
Config Configuration for supervised fine-tuning.
SupervisedDataset Abstract base for datasets used in supervised learning.
SupervisedDatasetBuilder A chz config class that knows how to construct a supervised dataset.
ChatDatasetBuilderCommonConfig Shared configuration for all chat-based dataset builders.
ChatDatasetBuilder Build a chat dataset that uses a renderer to tokenise message lists.
SupervisedDatasetFromHFDataset A supervised dataset backed by a HuggingFace dataset.
StreamingSupervisedDatasetFromHFDataset A supervised dataset that streams from HuggingFace, reducing memory usage.
FromConversationFileBuilder Build a supervised dataset from a JSONL file of chat conversations.