Update sam2/modeling/sam/transformer.py

sam2/modeling/sam/transformer.py

@@ -17,6 +17,7 @@ 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
+from einops import rearrange 
 
 warnings.simplefilter(action="ignore", category=FutureWarning)
 # OLD_GPU, USE_FLASH_ATTN, MATH_KERNEL_ON = get_sdpa_settings()