mindformers.generation

GenerationConfig

mindformers.generation.GenerationConfig

Class that holds a configuration for a generation task.

GenerationMixin

mindformers.generation.GenerationMixin

A class providing all functions for autoregressive text generation, used as a mixin with PreTrainedModel.