fffiloni commited on
Commit
9b937eb
1 Parent(s): 201a9cc

Update sam2/modeling/sam/transformer.py

Browse files
Files changed (1) hide show
  1. sam2/modeling/sam/transformer.py +3 -1
sam2/modeling/sam/transformer.py CHANGED
@@ -19,7 +19,9 @@ from sam2.modeling.sam2_utils import MLP
19
  from sam2.utils.misc import get_sdpa_settings
20
 
21
  warnings.simplefilter(action="ignore", category=FutureWarning)
22
- OLD_GPU, USE_FLASH_ATTN, MATH_KERNEL_ON = get_sdpa_settings()
 
 
23
 
24
 
25
  class TwoWayTransformer(nn.Module):
 
19
  from sam2.utils.misc import get_sdpa_settings
20
 
21
  warnings.simplefilter(action="ignore", category=FutureWarning)
22
+ USE_FLASH_ATTN = False
23
+ MATH_KERNEL_ON = True
24
+ OLD_GPU = True
25
 
26
 
27
  class TwoWayTransformer(nn.Module):