update for GPU usage

annotator/hed/__init__.py

@@ -87,10 +87,11 @@ class SOFT_HEDdetector:
         if not os.path.exists(modelpath):
             from basicsr.utils.download_util import load_file_from_url
             load_file_from_url(remote_model_path, model_dir=annotator_ckpts_path)
-        self.netNetwork = ControlNetHED_Apache2().float().cuda().eval()
-        self.netNetwork.load_state_dict(torch.load(modelpath))
+        self.netNetwork = ControlNetHED_Apache2().float().eval()
+        self.netNetwork.load_state_dict(torch.load(modelpath, map_location='cpu'))
 
     def __call__(self, input_image, safe=False, threshold=200):
+        self.netNetwork.cuda()
         assert input_image.ndim == 3
         H, W, C = input_image.shape
         with torch.no_grad():