This commit is contained in:
Manuel Romero 2021-02-13 14:26:01 +01:00 коммит произвёл GitHub
Родитель c9837a0d27
Коммит c969366870
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: 4AEE18F83AFDEB23
1 изменённых файлов: 2 добавлений и 2 удалений

Просмотреть файл

@ -1245,7 +1245,7 @@ class GenerationMixin:
outputs, model_kwargs, is_encoder_decoder=self.config.is_encoder_decoder
)
# 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 unfinished_sequences.max() == 0:
break
@ -1451,7 +1451,7 @@ class GenerationMixin:
sequence_lengths, unfinished_sequences, cur_len, next_tokens == eos_token_id
)
# 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 unfinished_sequences.max() == 0:
break