diff --git a/config.json b/config.json
index 593281a..f70148d 100644
--- a/config.json
+++ b/config.json
@@ -70,4 +70,10 @@
   "vocab_size": 50257,
   "window_size": 256,
   "tokenizer_class": "GPT2Tokenizer"
+  "task_specific_params": {
+    "text-generation": {
+      "do_sample": true,
+      "max_length": 50
+    }
+  }
 }