Update sam2/modeling/sam/transformer.py

sam2/modeling/sam/transformer.py

@@ -16,6 +16,7 @@ from torch import Tensor, nn
 from sam2.modeling.position_encoding import apply_rotary_enc, compute_axial_cis
 from sam2.modeling.sam2_utils import MLP
 from sam2.utils.misc import get_sdp_backends
+import flash_attn_interface
 
 warnings.simplefilter(action="ignore", category=FutureWarning)
 # OLD_GPU, USE_FLASH_ATTN, MATH_KERNEL_ON = get_sdpa_settings()