Spaces:

symanto
/

generation_evaluator

Sleeping

jogonba2 commited on Sep 5, 2024

Commit

3445e81

1 Parent(s): f40bceb

revert to gpu

Files changed (1) hide show

generation_evaluator.py CHANGED Viewed

@@ -133,7 +133,7 @@ CHRF:{
 ALIGNSCORE_ARGS = {
     "model": "roberta-base",
     "batch_size": 32,
-    "device": "cpu",
     "ckpt_path": "https://huggingface.co/yzha/AlignScore/resolve/main/AlignScore-base.ckpt",
     "evaluation_mode": "nli_sp",
 }

 ALIGNSCORE_ARGS = {
     "model": "roberta-base",
     "batch_size": 32,
+    "device": "cuda",
     "ckpt_path": "https://huggingface.co/yzha/AlignScore/resolve/main/AlignScore-base.ckpt",
     "evaluation_mode": "nli_sp",
 }