From 2b8b6c929e282958a920ba2aa26ee59106986ec3 Mon Sep 17 00:00:00 2001 From: Sam Shleifer Date: Thu, 4 Jun 2020 08:13:52 -0400 Subject: [PATCH] [cleanup] PretrainedModel.generate: remove unused kwargs (#4761) --- src/transformers/modeling_utils.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index b557aaf8d3..edd11b174d 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -1143,9 +1143,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin): repetition_penalty=repetition_penalty, no_repeat_ngram_size=no_repeat_ngram_size, bad_words_ids=bad_words_ids, - bos_token_id=bos_token_id, pad_token_id=pad_token_id, - decoder_start_token_id=decoder_start_token_id, eos_token_id=eos_token_id, batch_size=effective_batch_size, num_return_sequences=num_return_sequences, @@ -1170,9 +1168,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin): repetition_penalty=repetition_penalty, no_repeat_ngram_size=no_repeat_ngram_size, bad_words_ids=bad_words_ids, - bos_token_id=bos_token_id, pad_token_id=pad_token_id, - decoder_start_token_id=decoder_start_token_id, eos_token_id=eos_token_id, batch_size=effective_batch_size, encoder_outputs=encoder_outputs, @@ -1196,10 +1192,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin): repetition_penalty, no_repeat_ngram_size, bad_words_ids, - bos_token_id, pad_token_id, eos_token_id, - decoder_start_token_id, batch_size, encoder_outputs, attention_mask, @@ -1318,10 +1312,8 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin): repetition_penalty, no_repeat_ngram_size, bad_words_ids, - bos_token_id, pad_token_id, eos_token_id, - decoder_start_token_id, batch_size, num_return_sequences, length_penalty,