Update mask_adapter/sam_maskadapter.py

mask_adapter/sam_maskadapter.py

@@ -79,7 +79,7 @@ class SAMVisualizationDemo(object):
                                                 stability_score_thresh=0.7,
                                                 crop_n_layers=0,
                                                 crop_n_points_downscale_factor=2,
-                                                min_mask_region_area=100).to(self.cpu_device)
+                                                min_mask_region_area=100)
 
         self.clip_model = clip_model.to(self.cpu_device)
         self.mask_adapter = mask_adapter.to(self.cpu_device)
@@ -220,7 +220,7 @@ class SAMPointVisualizationDemo(object):
 
         self.sam2 = sam2
 
-        self.predictor = SAM2ImagePredictor(sam2).to(self.cpu_device)
+        self.predictor = SAM2ImagePredictor(sam2)
 
         self.clip_model = clip_model.to(self.cpu_device)