mipatov commited on
Commit
f72f3a0
·
1 Parent(s): 2a8c266

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -31,8 +31,6 @@ def predict_gpt(text, model, tokenizer, temperature=1.0):
31
  max_length=512,
32
  eos_token_id = tokenizer.eos_token_id,
33
  pad_token_id = tokenizer.pad_token_id,
34
- repetition_penalty = 2.5,
35
- length_penalty = 0.7,
36
  num_return_sequences = 1,
37
  output_attentions = True,
38
  return_dict_in_generate=True,
 
31
  max_length=512,
32
  eos_token_id = tokenizer.eos_token_id,
33
  pad_token_id = tokenizer.pad_token_id,
 
 
34
  num_return_sequences = 1,
35
  output_attentions = True,
36
  return_dict_in_generate=True,