Update config.json
Browse files- config.json +6 -5
config.json
CHANGED
@@ -46,14 +46,15 @@
|
|
46 |
"use_cache": true,
|
47 |
"vocab_size": 63999,
|
48 |
"window_size": 256,
|
49 |
-
"task_specific_params": {
|
50 |
-
|
|
|
51 |
"do_sample": true,
|
52 |
"max_length": 50,
|
53 |
"num_beams": 5,
|
54 |
"top_p": 0.95,
|
55 |
"repetition_penalty": 3.0,
|
56 |
-
"no_repeat_ngram_size": 3
|
57 |
-
|
58 |
-
|
59 |
}
|
|
|
46 |
"use_cache": true,
|
47 |
"vocab_size": 63999,
|
48 |
"window_size": 256,
|
49 |
+
"task_specific_params": {
|
50 |
+
"text-generation":
|
51 |
+
{
|
52 |
"do_sample": true,
|
53 |
"max_length": 50,
|
54 |
"num_beams": 5,
|
55 |
"top_p": 0.95,
|
56 |
"repetition_penalty": 3.0,
|
57 |
+
"no_repeat_ngram_size": 3
|
58 |
+
}
|
59 |
+
}
|
60 |
}
|