Skip to content

tinker_cookbook.distillation.PromptOnlyDatasetBuilder

class tinker_cookbook.distillation.PromptOnlyDatasetBuilder(RLDatasetBuilder)

Builder for prompt-only datasets.

Fields:

  • dataset_name (str) – e.g., "deepmath"
  • groups_per_batch (int)
  • group_size (int)
  • model_name_for_tokenizer (str)
  • renderer_name (str)
  • convo_prefix (list[renderers.Message] | None) – Default: None
  • max_prompt_tokens (int | None) – Maximum tokens per prompt (None = no truncation) Default: 1024.