update

src/backend/tasks/cnndm/task_v2.py

@@ -100,12 +100,16 @@ class CNNDMv2(ConfigurableTask):
     def test_docs(self):
         return self.dataset["test"]
 
-    def custom_prompt(self):
-        res = "Provide a summary of the provided article."
-        return res
+    # def custom_prompt(self):
+    #     res = "Provide a summary of the provided article."
+    #     return res
+
+    # def fewshot_delimiter(self):
+    #     return "\n\n"
 
+    # From https://arxiv.org/abs/2305.14739
     def doc_to_text(self, doc):
-        return f'{self.custom_prompt()}\n\nArticle: {doc["article"]}\nSummary:'
+        return f'Article: {doc["article"]}\nSummarize the article. Summary:'
 
     @staticmethod
     def should_decontaminate():

src/backend/tasks/xsum/task_v2.py

@@ -92,12 +92,12 @@ class XSumv2(ConfigurableTask):
     def test_docs(self):
         return self.dataset["test"]
 
-    def custom_prompt(self):
-        res = "Provide a summary of the provided document."
-        return res
+    # def fewshot_delimiter(self):
+    #     return "\n\n"
 
+    # From https://arxiv.org/abs/2305.14739
     def doc_to_text(self, doc):
-        return f'{self.custom_prompt()}\n\nDocument: {doc["document"]}\nSummary:'
+        return f'Article: {doc["document"]}\nSummarize the article in one sentence. Summary:'
 
     def should_decontaminate(self):
         return True