AlpacaToMessages

class torchtune.data.AlpacaToMessages(train_on_input: bool = True, column_map: Optional[Dict[str, str]] = None)

用于 Alpaca 风格数据集的消息转换类，数据集包含 “instruction”、“input” 和 “output” 列（或 column_map 中指定的等效字段）。用户消息由 instruction + input 列构成，助手消息由 output 列构成。提示模板化取决于 “input” 列是否存在，因此直接在此转换类中处理，而不是在专用的 PromptTemplate 类中处理，因为存在这种自定义逻辑。

参数:

• train_on_input (bool) – 模型是否在用户提示上进行训练。默认为 True。

• column_map (Optional[Dict[str, str]]) – 用于将预期的 “instruction”、“input” 和 “output” 列名更改为数据集中实际列名的映射。默认为 None，保留默认列名。