changed to text_generation task

config.json

@@ -22,32 +22,14 @@
   "pad_token_id": 0,
   "relative_attention_num_buckets": 32,
   "task_specific_params": {
-    "summarization": {
+    "text_generation": {
       "early_stopping": true,
       "length_penalty": 2.0,
-      "max_length": 200,
+      "max_length": 500,
       "min_length": 30,
       "no_repeat_ngram_size": 3,
       "num_beams": 4,
-      "prefix": "summarize: "
-    },
-    "translation_en_to_de": {
-      "early_stopping": true,
-      "max_length": 300,
-      "num_beams": 4,
-      "prefix": "translate English to German: "
-    },
-    "translation_en_to_fr": {
-      "early_stopping": true,
-      "max_length": 300,
-      "num_beams": 4,
-      "prefix": "translate English to French: "
-    },
-    "translation_en_to_ro": {
-      "early_stopping": true,
-      "max_length": 300,
-      "num_beams": 4,
-      "prefix": "translate English to Romanian: "
+      "prefix": ""
     }
   },
   "torch_dtype": "float32",