Updated config for different outputs

config.json

@@ -28,5 +28,11 @@
   "torch_dtype": "float32",
   "transformers_version": "4.17.0",
   "use_cache": true,
+  "task_specific_params": {
+    "text-generation": {
+      "do_sample": true,
+      "max_length": 50
+    }
+  },
   "vocab_size": 50257
 }