Unverified Commit da37a21c authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

push (#9008)

parent 61abd50b
...@@ -15,7 +15,8 @@ Utilities for Generation ...@@ -15,7 +15,8 @@ Utilities for Generation
This page lists all the utility functions used by :meth:`~transformers.PretrainedModel.generate`, This page lists all the utility functions used by :meth:`~transformers.PretrainedModel.generate`,
:meth:`~transformers.PretrainedModel.greedy_search`, :meth:`~transformers.PretrainedModel.sample`, :meth:`~transformers.PretrainedModel.greedy_search`, :meth:`~transformers.PretrainedModel.sample`,
:meth:`~transformers.PretrainedModel.beam_search`, and :meth:`~transformers.PretrainedModel.beam_sample`. :meth:`~transformers.PretrainedModel.beam_search`, :meth:`~transformers.PretrainedModel.beam_sample`, and
:meth:`~transformers.PretrainedModel.group_beam_search`.
Most of those are only useful if you are studying the code of the generate methods in the library. Most of those are only useful if you are studying the code of the generate methods in the library.
......
...@@ -415,8 +415,8 @@ class PrefixConstrainedLogitsProcessor(LogitsProcessor): ...@@ -415,8 +415,8 @@ class PrefixConstrainedLogitsProcessor(LogitsProcessor):
class HammingDiversityLogitsProcessor(LogitsProcessor): class HammingDiversityLogitsProcessor(LogitsProcessor):
r""" r"""
:class:`transformers.LogitsProcessor` that enforces diverse beam search. Note that this logits processor is only :class:`transformers.LogitsProcessor` that enforces diverse beam search. Note that this logits processor is only
effective for `group_beam_search`. See `Diverse Beam Search: Decoding Diverse Solutions from Neural Sequence Models effective for :meth:`transformers.PretrainedModel.group_beam_search`. See `Diverse Beam Search: Decoding Diverse
<https://arxiv.org/pdf/1610.02424.pdf>`__ for more details. Solutions from Neural Sequence Models <https://arxiv.org/pdf/1610.02424.pdf>`__ for more details.
Args: Args:
diversity_penalty (:obj:`float`): diversity_penalty (:obj:`float`):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment