From f0329ea516d1562c9e7442b87e5eb76d2b876eae Mon Sep 17 00:00:00 2001 From: tomohideshibata Date: Wed, 27 Jan 2021 19:15:23 +0900 Subject: [PATCH] Delete a needless duplicate condition (#9826) Co-authored-by: Tomohide Shibata --- src/transformers/generation_logits_process.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/transformers/generation_logits_process.py b/src/transformers/generation_logits_process.py index a027eacbde..ac77b6b224 100644 --- a/src/transformers/generation_logits_process.py +++ b/src/transformers/generation_logits_process.py @@ -439,8 +439,6 @@ class HammingDiversityLogitsProcessor(LogitsProcessor): raise ValueError("`num_beam_groups` should be an integer strictly larger than 1.") if num_beam_groups > num_beams: raise ValueError("`beam_groups` has to be smaller or equal to `num_beams`.") - if num_beam_groups > num_beams: - raise ValueError("`beam_groups` has to be smaller or equal to `num_beams`") self._num_sub_beams = num_beams // num_beam_groups def __call__(