text_generation

text_generation#

class TextGenerationTask(task, model, **kwargs)[source]#

Bases: Task

The text generation model to predict the question or chinese poetry. :param task: The name of task. :type task: string :param model: The model name in the task. :type model: string :param kwargs: Additional keyword arguments passed along to the specific task. :type kwargs: dict, optional