tinker_cookbook.renderers.TrainOnWhat

class tinker_cookbook.renderers.TrainOnWhat( StrEnum)

Enum controlling which parts of the sequence to compute loss on.

Members:

• LAST_ASSISTANT_MESSAGE: Train only on the final assistant message.
• LAST_ASSISTANT_TURN: Train on the last assistant turn (including any tool calls and tool responses leading to the final assistant reply).
• ALL_ASSISTANT_MESSAGES: Train on every assistant message in the conversation.
• ALL_MESSAGES: Train on all messages regardless of role.
• ALL_TOKENS: Train on every token in the sequence (including special tokens).
• ALL_USER_AND_SYSTEM_MESSAGES: Train on user and system messages only.
• CUSTOMIZED: Use per-message train flags from the dataset.

Referenced by

• tinker_cookbook.renderers.Renderer.build_supervised_example
• tinker_cookbook.renderers.Renderer.build_supervised_examples
• tinker_cookbook.supervised.ChatDatasetBuilderCommonConfig