SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Delete a needless duplicate condition (#9826)

Browse Source 
Co-authored-by: Tomohide Shibata <tomshiba@yahoo-corp.jp>
This commit is contained in:
tomohideshibata
2021-01-27 19:15:23 +09:00
committed by GitHub
parent a1720694a5
commit f0329ea516
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/generation_logits_process.py
View File

@@ -439,8 +439,6 @@ class HammingDiversityLogitsProcessor(LogitsProcessor):
raise ValueError("`num_beam_groups` should be an integer strictly larger than 1.") raise ValueError("`num_beam_groups` should be an integer strictly larger than 1.")
if num_beam_groups > num_beams: if num_beam_groups > num_beams:
raise ValueError("`beam_groups` has to be smaller or equal to `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 self._num_sub_beams = num_beams // num_beam_groups
def __call__( def __call__(