PyTorch | |
[[autodoc]] generation.GenerateDecoderOnlyOutput | |
[[autodoc]] generation.GenerateEncoderDecoderOutput | |
[[autodoc]] generation.GenerateBeamDecoderOnlyOutput | |
[[autodoc]] generation.GenerateBeamEncoderDecoderOutput | |
TensorFlow | |
[[autodoc]] generation.TFGreedySearchEncoderDecoderOutput | |
[[autodoc]] generation.TFGreedySearchDecoderOnlyOutput | |
[[autodoc]] generation.TFSampleEncoderDecoderOutput | |
[[autodoc]] generation.TFSampleDecoderOnlyOutput | |
[[autodoc]] generation.TFBeamSearchEncoderDecoderOutput | |
[[autodoc]] generation.TFBeamSearchDecoderOnlyOutput | |
[[autodoc]] generation.TFBeamSampleEncoderDecoderOutput | |
[[autodoc]] generation.TFBeamSampleDecoderOnlyOutput | |
[[autodoc]] generation.TFContrastiveSearchEncoderDecoderOutput | |
[[autodoc]] generation.TFContrastiveSearchDecoderOnlyOutput | |
FLAX | |
[[autodoc]] generation.FlaxSampleOutput | |
[[autodoc]] generation.FlaxGreedySearchOutput | |
[[autodoc]] generation.FlaxBeamSearchOutput | |
LogitsProcessor | |
A [LogitsProcessor] can be used to modify the prediction scores of a language model head for | |
generation. |