From 822f725a07fda22042bb50b71c66be83007a9d1c Mon Sep 17 00:00:00 2001 From: patrickvonplaten Date: Mon, 23 Dec 2019 21:49:32 +0100 Subject: [PATCH] duplicated line for repeating_words_penalty_for_language_generation --- src/transformers/modeling_utils.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py index 8722d578fd..5ae89ff691 100644 --- a/src/transformers/modeling_utils.py +++ b/src/transformers/modeling_utils.py @@ -585,8 +585,6 @@ class PreTrainedModel(nn.Module): Beginning of sentence token if no prompt is provided. Default to 0. **eos_token_ids**: (`optional`) int or list of int End of sequence token or list of tokens to stop the generation. Default to 0. - **length_penalty**: (`optional`) int - Exponential penalty to the length. Default to 0. **length_penalty**: (`optional`) float Exponential penalty to the length. Default to 1. **num_return_sequences**: (`optional`) int