Added bad_words_ids
Browse files- generation_config.json +15 -16
generation_config.json
CHANGED
@@ -1,16 +1,15 @@
|
|
1 |
-
{
|
2 |
-
"bos_token_id": 50256,
|
3 |
-
"
|
4 |
-
"
|
5 |
-
"
|
6 |
-
"
|
7 |
-
"
|
8 |
-
"
|
9 |
-
"
|
10 |
-
"
|
11 |
-
"
|
12 |
-
"seed": 12,
|
13 |
-
"temperature": 0.8,
|
14 |
-
"top_p": 0.9
|
15 |
-
|
16 |
-
}
|
|
|
1 |
+
{
|
2 |
+
"bos_token_id": 50256,
|
3 |
+
"eos_token_id": 50256,
|
4 |
+
"pad_token_id": 50256,
|
5 |
+
"max_length": 50,
|
6 |
+
"no_repeat_ngram_size": 3,
|
7 |
+
"num_beams": 4,
|
8 |
+
"max_time": 5,
|
9 |
+
"early_stopping": true,
|
10 |
+
"repetition_penalty": 1.5,
|
11 |
+
"length_penalty": 1.7,
|
12 |
+
"seed": 12,
|
13 |
+
"temperature": 0.8,
|
14 |
+
"top_p": 0.9
|
15 |
+
}
|
|