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
 }