From da37a21c89c0555b6002b1fd51ee094e35a2e942 Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 9 Dec 2020 15:14:33 +0100 Subject: [PATCH] push (#9008) --- docs/source/internal/generation_utils.rst | 3 ++- src/transformers/generation_logits_process.py | 4 ++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/docs/source/internal/generation_utils.rst b/docs/source/internal/generation_utils.rst index 6851cb7299..c608a75f5e 100644 --- a/docs/source/internal/generation_utils.rst +++ b/docs/source/internal/generation_utils.rst @@ -15,7 +15,8 @@ Utilities for Generation 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.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. diff --git a/src/transformers/generation_logits_process.py b/src/transformers/generation_logits_process.py index 8ffab62386..465cdc4f6d 100644 --- a/src/transformers/generation_logits_process.py +++ b/src/transformers/generation_logits_process.py @@ -415,8 +415,8 @@ class PrefixConstrainedLogitsProcessor(LogitsProcessor): class HammingDiversityLogitsProcessor(LogitsProcessor): r""" :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 - `__ for more details. + effective for :meth:`transformers.PretrainedModel.group_beam_search`. See `Diverse Beam Search: Decoding Diverse + Solutions from Neural Sequence Models `__ for more details. Args: diversity_penalty (:obj:`float`):