Update setup.py, installation instructions and transformer settings

README.md

@@ -17,9 +17,7 @@
 Please install SAM 2 on a GPU machine using:
 
 ```bash
-git clone https://github.com/facebookresearch/segment-anything-2.git
-
-cd segment-anything-2; pip install -e .
+pip install git+https://github.com/facebookresearch/segment-anything-2.git
 ```
 
 To use the SAM 2 predictor and run the example notebooks, `jupyter` and `matplotlib` are required and can be installed by:

sam2/modeling/sam/transformer.py

@@ -21,6 +21,10 @@
 warnings.simplefilter(action="ignore", category=FutureWarning)
 OLD_GPU, USE_FLASH_ATTN, MATH_KERNEL_ON = get_sdpa_settings()
 
+USE_FLASH_ATTN = False
+MATH_KERNEL_ON = True
+OLD_GPU = True
+
 
 class TwoWayTransformer(nn.Module):
     def __init__(

setup.py

@@ -69,4 +69,8 @@ def get_extensions():
     python_requires=">=3.10.0",
     ext_modules=get_extensions(),
     cmdclass={"build_ext": BuildExtension.with_options(no_python_abi_suffix=True)},
+    include_package_data=True,
+    package_data={
+        "sam2_configs": ["*.yaml"],
+    },
 )