Update config.json
Browse files- config.json +7 -2
config.json
CHANGED
@@ -68,8 +68,13 @@
|
|
68 |
"task_specific_params": {
|
69 |
"text-generation": {
|
70 |
"do_sample": true,
|
71 |
-
"max_length":
|
72 |
-
"temperature": 0.
|
|
|
|
|
|
|
|
|
|
|
73 |
}
|
74 |
},
|
75 |
"tokenizer_class": "GPT2Tokenizer",
|
|
|
68 |
"task_specific_params": {
|
69 |
"text-generation": {
|
70 |
"do_sample": true,
|
71 |
+
"max_length": 2048,
|
72 |
+
"temperature": 0.5,
|
73 |
+
"top_p" = 0.9,
|
74 |
+
"top_k" = 0,
|
75 |
+
"repetition_penalty" = 1.1,
|
76 |
+
"pad_token_id" = 50256,
|
77 |
+
"num_return_sequences" = 1
|
78 |
}
|
79 |
},
|
80 |
"tokenizer_class": "GPT2Tokenizer",
|