MahmoudH commited on
Commit
a427a6e
·
1 Parent(s): ec5ce8f

Decrease `length_penalty`

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ def predict(text):
11
  input_ids=tokenized_inputs["input_ids"],
12
  attention_mask=tokenized_inputs["attention_mask"],
13
  max_new_tokens=256,
14
- length_penalty=0.8,
15
  num_beams=4,
16
  do_sample=True
17
  )
 
11
  input_ids=tokenized_inputs["input_ids"],
12
  attention_mask=tokenized_inputs["attention_mask"],
13
  max_new_tokens=256,
14
+ length_penalty=0.5,
15
  num_beams=4,
16
  do_sample=True
17
  )