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

[cleanup] PretrainedModel.generate: remove unused kwargs (#4761)

Browse Source
This commit is contained in:
Sam Shleifer
2020-06-04 08:13:52 -04:00
committed by GitHub
parent 5bf9afbf35
commit 2b8b6c929e
1 changed files with 0 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/transformers/modeling_utils.py
View File

@@ -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,