pszemraj commited on
Commit
3ec7db9
1 Parent(s): 32e1caf

update inference params

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -52,12 +52,12 @@ Nearby tokens should be important because, in a sentence (sequence of words), th
52
 
53
  inference:
54
  parameters:
55
- max_length: 96
56
- min_length: 4
57
  no_repeat_ngram_size: 3
58
  early_stopping: True
59
- repetition_penalty: 2.4
60
- length_penalty: 0.3
61
  encoder_no_repeat_ngram_size : 4
62
  num_beams : 4
63
 
 
52
 
53
  inference:
54
  parameters:
55
+ max_length: 64
56
+ min_length: 8
57
  no_repeat_ngram_size: 3
58
  early_stopping: True
59
+ repetition_penalty: 3.5
60
+ length_penalty: 0.1
61
  encoder_no_repeat_ngram_size : 4
62
  num_beams : 4
63