pminervini commited on
Commit
bc4faee
1 Parent(s): 210a90b
Files changed (1) hide show
  1. src/backend/tasks/cnndm/task_v2.py +2 -2
src/backend/tasks/cnndm/task_v2.py CHANGED
@@ -100,12 +100,12 @@ class CNNDMv2(ConfigurableTask):
100
  def test_docs(self):
101
  return self.dataset["test"]
102
 
103
- def prompt(self):
104
  res = "Provide a summary of the provided article."
105
  return res
106
 
107
  def doc_to_text(self, doc):
108
- return f'{self.prompt()}\n\nArticle: {doc["article"]}\nSummary:'
109
 
110
  @staticmethod
111
  def should_decontaminate():
 
100
  def test_docs(self):
101
  return self.dataset["test"]
102
 
103
+ def custom_prompt(self):
104
  res = "Provide a summary of the provided article."
105
  return res
106
 
107
  def doc_to_text(self, doc):
108
+ return f'{self.custom_prompt()}\n\nArticle: {doc["article"]}\nSummary:'
109
 
110
  @staticmethod
111
  def should_decontaminate():