works running locally

Author: micaelakaplan

label_studio_ml/examples/segment_anything_2_image/model.py

@@ -17,7 +17,7 @@
 from sam2.sam2_image_predictor import SAM2ImagePredictor
 
 
-DEVICE = os.getenv('DEVICE', 'cuda')
+DEVICE = os.getenv('DEVICE', 'cpu')
 MODEL_CONFIG = os.getenv('MODEL_CONFIG', 'configs/sam2.1/sam2.1_hiera_l.yaml')
 MODEL_CHECKPOINT = os.getenv('MODEL_CHECKPOINT', 'sam2.1_hiera_large.pt')