Update config.json

config.json

@@ -30,6 +30,14 @@
     "text-generation": {
       "do_sample": true,
       "max_length": 50
+    },
+    "c2gen": {
+      "prompt_template" : "You are a context generator that generates 'CONTEXT' based on the given 'KEYWORDS'.\n\nKEYWORDS: \n{keywords}\n\nCONTEXT: \n{context}\n\n:END".format(keywords="{keywords}", context="{context}"),
+      "do_sample": true,
+      "max_length": 256,
+      "temperature": 0.7,
+      "top_k": 32,
+      "top_p": 0.88,
     }
   },
   "torch_dtype": "float32",