fix sam hq and mobile sam loading with non cuda system (#190)

Author: light-and-ray

.gitignore

@@ -1,3 +1,4 @@
 models/*/*.pth
+models/*/*.pt
 __pycache__
-annotator
+annotator

sam_hq/build_sam_hq.py

@@ -64,7 +64,7 @@ def _load_sam_checkpoint(sam: Sam, checkpoint=None):
     sam.eval()
     if checkpoint is not None:
         with open(checkpoint, "rb") as f:
-            state_dict = torch.load(f)
+            state_dict = torch.load(f, map_location="cpu")
         info = sam.load_state_dict(state_dict, strict=False)
         print(info)
     for _, p in sam.named_parameters():