pszemraj commited on
Commit
4299ec7
1 Parent(s): 9a0c01e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -24,10 +24,10 @@ inference:
24
  parameters:
25
  max_length: 64
26
  no_repeat_ngram_size: 3
27
- do_sample: True
28
- top_p: 0.95
29
- repetition_penalty: 3.7
30
- num_beams : 2
31
 
32
 
33
  ---
 
24
  parameters:
25
  max_length: 64
26
  no_repeat_ngram_size: 3
27
+ encoder_no_repeat_ngram_size: 3
28
+ repetition_penalty: 1.4
29
+ num_beams: 4
30
+ early_stopping: True
31
 
32
 
33
  ---