Update eval_wrapper/eval.py

eval_wrapper/eval.py

@@ -23,7 +23,7 @@ from eval_wrapper.eval_utils import filter_all_masks
 from huggingface_hub import hf_hub_download
 
 class EvalWrapper(torch.nn.Module):
-    def __init__(self,checkpoint_path,distributed=False,device="cuda",dtype=torch.float32,**kwargs):
+    def __init__(self,checkpoint_path,distributed=False,device="cpu",dtype=torch.float32,**kwargs):
         super().__init__()
         checkpoint = torch.load(checkpoint_path, map_location='cpu',weights_only=False)
         model_string = checkpoint['args'].model