pminervini commited on
Commit
7fe1886
1 Parent(s): 10f3d68
Files changed (1) hide show
  1. src/backend/tasks/xsum/task.py +1 -1
src/backend/tasks/xsum/task.py CHANGED
@@ -152,7 +152,7 @@ class XSum(Task):
152
  factkb_logits = self.factkb_model(**factkb_tokens).logits
153
  factkb_res = torch.softmax(factkb_logits, dim=1)
154
 
155
- self.maybe_init_factkb()
156
  bert_score_res = self.bert_score.compute(predictions=[completion], references=[gold_summary], lang="en")
157
 
158
  res = {
 
152
  factkb_logits = self.factkb_model(**factkb_tokens).logits
153
  factkb_res = torch.softmax(factkb_logits, dim=1)
154
 
155
+ self.maybe_init_bertscore()
156
  bert_score_res = self.bert_score.compute(predictions=[completion], references=[gold_summary], lang="en")
157
 
158
  res = {