Peter commited on
Commit
3247bd6
1 Parent(s): 504e8b4

update gen params

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -68,6 +68,9 @@ def proc_submission(
68
  "no_repeat_ngram_size": no_repeat_ngram_size,
69
  "encoder_no_repeat_ngram_size": 4,
70
  "num_beams": num_beams,
 
 
 
71
  }
72
 
73
  history = {}
 
68
  "no_repeat_ngram_size": no_repeat_ngram_size,
69
  "encoder_no_repeat_ngram_size": 4,
70
  "num_beams": num_beams,
71
+ "early_stopping": True,
72
+ "min_length": 10,
73
+ "do_sample": False,
74
  }
75
 
76
  history = {}