fix sam2

backbone.py

@@ -40,7 +40,7 @@ class SAM2(nn.Module):
                 f.write(r.content)
         sam2_checkpoint = filename
         
-        device = 'cuda' if torch.cuda.is_available() else 'cpu'
+        device = 'cpu'
         sam2_model = build_sam2(model_cfg, sam2_checkpoint, device=device)
 
         image_encoder = sam2_model.image_encoder
@@ -835,7 +835,11 @@ RES_DICT["ImageNet(vit_base)"] = (672, 672)
 def download_all_models():
     for model_name in MODEL_DICT:
         print(f"Downloading {model_name}")
-        model = MODEL_DICT[model_name]()
+        try:
+            model = MODEL_DICT[model_name]()
+        except Exception as e:
+            print(f"Error downloading {model_name}: {e}")
+            continue
         
 def get_all_model_names():
     return list(MODEL_DICT.keys())