pszemraj commited on
Commit
4566061
1 Parent(s): f133f0b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -19,10 +19,11 @@ widget:
19
  inference:
20
  parameters:
21
  max_length: 256
22
- do_sample: True
23
- top_p: 0.95
24
  repetition_penalty: 1.3
25
- num_beams : 2
 
26
  ---
27
 
28
  # pegasus does math?
 
19
  inference:
20
  parameters:
21
  max_length: 256
22
+ no_repeat_ngram_size: 4
23
+ length_penalty: 0.7
24
  repetition_penalty: 1.3
25
+ num_beams : 4
26
+ early_stopping: True
27
  ---
28
 
29
  # pegasus does math?