diff --git a/config.json b/config.json index ca0fb1d..fed02aa 100644 --- a/config.json +++ b/config.json @@ -28,5 +28,11 @@ "summary_proj_to_labels": true, "summary_type": "cls_index", "summary_use_proj": true, + "task_specific_params": { + "text-generation": { + "do_sample": true, + "max_length": 50 + } + }, "vocab_size": 50257 -} +} \ No newline at end of file