Commit 822f725a authored by patrickvonplaten's avatar patrickvonplaten
Browse files

duplicated line for repeating_words_penalty_for_language_generation

parent aeef4823
...@@ -585,8 +585,6 @@ class PreTrainedModel(nn.Module): ...@@ -585,8 +585,6 @@ class PreTrainedModel(nn.Module):
Beginning of sentence token if no prompt is provided. Default to 0. Beginning of sentence token if no prompt is provided. Default to 0.
**eos_token_ids**: (`optional`) int or list of int **eos_token_ids**: (`optional`) int or list of int
End of sequence token or list of tokens to stop the generation. Default to 0. 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 **length_penalty**: (`optional`) float
Exponential penalty to the length. Default to 1. Exponential penalty to the length. Default to 1.
**num_return_sequences**: (`optional`) int **num_return_sequences**: (`optional`) int
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment