update

src/backend/tasks/cnndm/task_v2.py

@@ -61,12 +61,13 @@ def rouge(refs, preds):
 
 # @register_task("cnndm_v2")
 class CNNDMv2(ConfigurableTask):
-    VERSION = 0
+    VERSION = 2
     DATASET_PATH = "cnn_dailymail"
     DATASET_NAME = "3.0.0"
 
     def __init__(self):
-        super().__init__(config={'metadata': {'version': self.VERSION}})
+        super().__init__(config={'metadata': {'version': self.VERSION},
+                                 'generation_kwargs': {'do_sample': False, 'temperature': 0.0, 'until': ['\n', '\n\n']}})
         self.factkb_tokenizer = None
         self.factkb_model = None
         self.bert_score = None

src/backend/tasks/xsum/task_v2.py

@@ -53,12 +53,14 @@ def rouge(refs, preds):
 
 # @register_task("xsum_v2")
 class XSumv2(ConfigurableTask):
-    VERSION = 0
+    VERSION = 2
     DATASET_PATH = "EdinburghNLP/xsum"
     DATASET_NAME = None
 
     def __init__(self):
-        super().__init__(config={'metadata': {'version': self.VERSION}})
+        # breakpoint()
+        super().__init__(config={'metadata': {'version': self.VERSION},
+                                 'generation_kwargs': {'do_sample': False, 'temperature': 0.0, 'until': ['\n', '\n\n']}})
         self.factkb_tokenizer = None
         self.factkb_model = None
         self.bert_score = None