Skip to content

Latest commit

聽

History

History
41 lines (29 loc) 路 1.46 KB

text_generation.mdx

File metadata and controls

41 lines (29 loc) 路 1.46 KB
Raw

Generation

Each framework has a generate method for auto-regressive text generation implemented in their respective GenerationMixin class:

  • PyTorch [~generation_utils.GenerationMixin.generate] is implemented in [~generation_utils.GenerationMixin].
  • TensorFlow [~generation_tf_utils.TFGenerationMixin.generate] is implemented in [~generation_tf_utils.TFGenerationMixin].
  • Flax/JAX [~generation_flax_utils.FlaxGenerationMixin.generate] is implemented in [~generation_flax_utils.FlaxGenerationMixin].

GenerationMixin

[[autodoc]] generation_utils.GenerationMixin - generate - greedy_search - sample - beam_search - beam_sample - contrastive_search - group_beam_search - constrained_beam_search

TFGenerationMixin

[[autodoc]] generation_tf_utils.TFGenerationMixin - generate

FlaxGenerationMixin

[[autodoc]] generation_flax_utils.FlaxGenerationMixin - generate