diff --git a/models/sam3/model/box_ops.py b/models/sam3/model/box_ops.py
index f88e4ad..c4318a3 100644
--- a/models/sam3/model/box_ops.py
+++ b/models/sam3/model/box_ops.py
@@ -142,7 +142,6 @@ def generalized_box_iou(boxes1, boxes2):
     return iou - (area - union) / area
 
 
-@torch.jit.script
 def fast_diag_generalized_box_iou(boxes1, boxes2):
     assert len(boxes1) == len(boxes2)
     box1_xy = boxes1[:, 2:]
@@ -168,8 +167,12 @@ def fast_diag_generalized_box_iou(boxes1, boxes2):
 
     return iou - (tot_area - union) / tot_area
 
+try:
+    fast_diag_generalized_box_iou = torch.jit.script(fast_diag_generalized_box_iou)
+except Exception:
+    pass
+
 
-@torch.jit.script
 def fast_diag_box_iou(boxes1, boxes2):
     assert len(boxes1) == len(boxes2)
     box1_xy = boxes1[:, 2:]
@@ -193,6 +196,12 @@ def fast_diag_box_iou(boxes1, boxes2):
     return iou
 
 
+try:
+    fast_diag_box_iou = torch.jit.script(fast_diag_box_iou)
+except Exception:
+    pass
+
+
 def box_xywh_inter_union(
     boxes1: torch.Tensor, boxes2: torch.Tensor
 ) -> Tuple[torch.Tensor, torch.Tensor]: