GeneratorTask

本节包含 distilabel 生成器任务的 API 参考。

有关 GeneratorTask 如何工作以及一些示例的更多信息，请查看 教程 - Task - GeneratorTask 页面。

GeneratorTask

基类: _Task, GeneratorStep

GeneratorTask 是一个类，它实现了 _Task 抽象类，并添加了 GeneratorStep 接口，以便在 pipeline 中用作 step。

属性

名称	类型	描述
llm	LLM	用于生成任务输出的 LLM。
group_generations	bool	是否将每个输入的 num_generations 生成结果分组在一个列表中，或者为每个生成结果创建一个行。默认为 False。
num_generations	RuntimeParameter[int]	每个输入要生成的数量。

源代码位于 src/distilabel/steps/tasks/base.py

class GeneratorTask(_Task, GeneratorStep):
    """`GeneratorTask` is a class that implements the `_Task` abstract class and adds the
    `GeneratorStep` interface to be used as a step in the pipeline.

    Attributes:
        llm: the `LLM` to be used to generate the outputs of the task.
        group_generations: whether to group the `num_generations` generated per input in
            a list or create a row per generation. Defaults to `False`.
        num_generations: The number of generations to be produced per input.
    """

    pass