Update README.md
Browse files
README.md
CHANGED
|
@@ -2,6 +2,9 @@
|
|
| 2 |
license: cc-by-nc-nd-4.0
|
| 3 |
inference:
|
| 4 |
parameters:
|
|
|
|
|
|
|
|
|
|
| 5 |
repetition_penalty: 10.0
|
| 6 |
diversity_penalty: 3.01
|
| 7 |
no_repeat_ngram_size: 2
|
|
|
|
| 2 |
license: cc-by-nc-nd-4.0
|
| 3 |
inference:
|
| 4 |
parameters:
|
| 5 |
+
num_beams: 1
|
| 6 |
+
num_beam_groups:
|
| 7 |
+
num_return_sequences: 1
|
| 8 |
repetition_penalty: 10.0
|
| 9 |
diversity_penalty: 3.01
|
| 10 |
no_repeat_ngram_size: 2
|