Update README.md
Browse files
README.md
CHANGED
@@ -19,10 +19,11 @@ widget:
|
|
19 |
inference:
|
20 |
parameters:
|
21 |
max_length: 256
|
22 |
-
|
23 |
-
|
24 |
repetition_penalty: 1.3
|
25 |
-
num_beams :
|
|
|
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?
|