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

Fix typo in comment (#10156)

Browse Source
This commit is contained in:
Manuel Romero
2021-02-13 14:26:25 +01:00
committed by GitHub
parent c969366870
commit 698c9e2dbd
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/generation_tf_utils.py
View File

@@ -551,7 +551,7 @@ class TFGenerationMixin:
# unfinished_sents is set to zero if eos in sentence # unfinished_sents is set to zero if eos in sentence
unfinished_sents -= is_sents_unfinished_and_token_to_add_is_eos unfinished_sents -= is_sents_unfinished_and_token_to_add_is_eos
# stop when there is a </s> in each sentence, or if we exceed the maximul length # stop when there is a </s> in each sentence, or if we exceed the maximum length
if tf.math.reduce_max(unfinished_sents) == 0: if tf.math.reduce_max(unfinished_sents) == 0:
break break