task_specific_params for text-generation

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "/content/drive/MyDrive/byt5-small/models/e3_lr0.0001",
+  "_name_or_path": "google/byt5-small",
   "architectures": [
     "T5ForConditionalGeneration"
   ],
@@ -24,5 +24,11 @@
   "torch_dtype": "float32",
   "transformers_version": "4.12.5",
   "use_cache": false,
+  "task_specific_params": {
+    "text-generation": {
+      "num_beams": 1,
+      "max_length": 64
+    }
+  },
   "vocab_size": 384
 }