Spaces:

yzha
/

ctc_eval

Runtime error

yzha commited on Jun 29, 2022

Commit

e251d64

1 Parent(s): db65786

update

Files changed (1) hide show

ctc_eval.py CHANGED Viewed

@@ -102,7 +102,7 @@ class CTC_Eval(evaluate.EvaluationModule):
         model_name, self.aspect = self.config_name.split(',')
         if self.aspect in ['consistency', 'relevance']:
-            self.scorer = SummarizationScorer(align=model_name)
         elif self.aspect in ['preservation']:
             self.scorer = StyleTransferScorer(align=model_name)
         elif self.aspect in ['engagingness', 'groundedness']:

         model_name, self.aspect = self.config_name.split(',')
         if self.aspect in ['consistency', 'relevance']:
+            self.scorer = SummarizationScorer(align=model_name, device='cpu')
         elif self.aspect in ['preservation']:
             self.scorer = StyleTransferScorer(align=model_name)
         elif self.aspect in ['engagingness', 'groundedness']: