Update eval/grounded_sam/grounded_sam2_florence2_autolabel_pipeline.py

eval/grounded_sam/grounded_sam2_florence2_autolabel_pipeline.py

@@ -60,7 +60,7 @@ class FlorenceSAM:
             self.torch_dtype = torch.bfloat16
             
         FLORENCE2_MODEL_ID = os.getenv('FLORENCE2_MODEL_PATH', "microsoft/Florence-2-large")
-        SAM2_CHECKPOINT = os.getenv('SAM2_MODEL_PATH')
+        SAM2_CHECKPOINT = os.getenv('SAM2_MODEL_PATH', "facebook/sam2-hiera-large")
         SAM2_CONFIG = "configs/sam2.1/sam2.1_hiera_l.yaml"
 
         self.florence2_model = Florence2ForConditionalGeneration.from_pretrained(
@@ -127,7 +127,7 @@ class FlorenceSAM:
 
     def segmentation(self, image, input_boxes, seg_model="sam"):
         if seg_model == "sam":
-            with torch.inference_mode(), torch.autocast("cuda", dtype=torch.float32):
+            with torch.inference_mode(), torch.autocast("cuda", dtype=torch.bfloat16):
                 sam2_predictor = self.sam2_predictor
                 sam2_predictor.set_image(np.array(image))
                 masks, scores, logits = sam2_predictor.predict(