Spaces:

X-iZhang
/

RadEval

Running

X-iZhang commited on Jul 13

Commit

73c92e0

verified ·

1 Parent(s): 9998f52

Update factual/green_score/green.py

Files changed (1) hide show

factual/green_score/green.py CHANGED Viewed

@@ -63,6 +63,7 @@ class GREEN:
             module="transformers.tokenization_utils_base",
         )
         cpu = cpu or not torch.cuda.is_available()
         self.cpu = cpu
         self.model_name = model_name.split("/")[-1]
         self.output_dir = output_dir

             module="transformers.tokenization_utils_base",
         )
         cpu = cpu or not torch.cuda.is_available()
+        self.device = torch.device("cpu" if cpu else "cuda")
         self.cpu = cpu
         self.model_name = model_name.split("/")[-1]
         self.output_dir = output_dir