Spaces:

fffiloni
/

SAM2-Image-Predictor

Sleeping

fffiloni commited on Aug 1

Commit

9b937eb

•

1 Parent(s): 201a9cc

Update sam2/modeling/sam/transformer.py

Files changed (1) hide show

sam2/modeling/sam/transformer.py CHANGED Viewed

@@ -19,7 +19,9 @@ from sam2.modeling.sam2_utils import MLP
 from sam2.utils.misc import get_sdpa_settings
 warnings.simplefilter(action="ignore", category=FutureWarning)
-OLD_GPU, USE_FLASH_ATTN, MATH_KERNEL_ON = get_sdpa_settings()
 class TwoWayTransformer(nn.Module):

 from sam2.utils.misc import get_sdpa_settings
 warnings.simplefilter(action="ignore", category=FutureWarning)
+USE_FLASH_ATTN = False
+MATH_KERNEL_ON = True
+OLD_GPU = True
 class TwoWayTransformer(nn.Module):