Spaces:
Runtime error
Runtime error
pminervini
commited on
Commit
•
7fe1886
1
Parent(s):
10f3d68
update
Browse files
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.
|
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 = {
|