José Ángel González commited on
Commit ·
3960e5f
1
Parent(s): 7c5780e
Update config.json
Browse filesParameters of `generate` in config
- config.json +8 -1
config.json
CHANGED
|
@@ -37,13 +37,20 @@
|
|
| 37 |
"task_specific_params": {
|
| 38 |
"summarization": {
|
| 39 |
"early_stopping": true,
|
| 40 |
-
"
|
| 41 |
"max_length": 150,
|
| 42 |
"min_length": 25,
|
| 43 |
"no_repeat_ngram_size": 3,
|
| 44 |
"num_beams": 4
|
| 45 |
}
|
| 46 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
"mask_sent_token_id": 6,
|
| 48 |
"mask_token_id": 5,
|
| 49 |
"max_position_embeddings": 512,
|
|
|
|
| 37 |
"task_specific_params": {
|
| 38 |
"summarization": {
|
| 39 |
"early_stopping": true,
|
| 40 |
+
"repetition_penalty": 2.0,
|
| 41 |
"max_length": 150,
|
| 42 |
"min_length": 25,
|
| 43 |
"no_repeat_ngram_size": 3,
|
| 44 |
"num_beams": 4
|
| 45 |
}
|
| 46 |
},
|
| 47 |
+
"early_stopping": true,
|
| 48 |
+
"repetition_penalty": 2.0,
|
| 49 |
+
"max_length": 150,
|
| 50 |
+
"min_length": 25,
|
| 51 |
+
"no_repeat_ngram_size": 3,
|
| 52 |
+
"num_beams": 4,
|
| 53 |
+
|
| 54 |
"mask_sent_token_id": 6,
|
| 55 |
"mask_token_id": 5,
|
| 56 |
"max_position_embeddings": 512,
|