Update generation_config.json
Browse files- generation_config.json +8 -8
generation_config.json
CHANGED
|
@@ -5,12 +5,12 @@
|
|
| 5 |
"eos_token_id": 2,
|
| 6 |
"pad_token_id": 2,
|
| 7 |
"max_length": 1024,
|
| 8 |
-
"num_beams"
|
| 9 |
-
"do_sample"
|
| 10 |
-
"early_stopping"
|
| 11 |
-
"temperature"
|
| 12 |
-
"top_k"
|
| 13 |
-
"top_p"
|
| 14 |
-
"repetition_penalty"
|
| 15 |
-
"no_repeat_ngram_size"
|
| 16 |
}
|
|
|
|
| 5 |
"eos_token_id": 2,
|
| 6 |
"pad_token_id": 2,
|
| 7 |
"max_length": 1024,
|
| 8 |
+
"num_beams":5,
|
| 9 |
+
"do_sample":True,
|
| 10 |
+
"early_stopping":True,
|
| 11 |
+
"temperature":0.7,
|
| 12 |
+
"top_k":50,
|
| 13 |
+
"top_p":0.95,
|
| 14 |
+
"repetition_penalty":1.2,
|
| 15 |
+
"no_repeat_ngram_size":3,
|
| 16 |
}
|