more reliable build

app.py

@@ -31,20 +31,24 @@ def sh(cmd): subprocess.check_call(cmd, shell=True)
 try:
 
     sh("pip install flash-attn --no-build-isolation")
+    # print("Attempting to download and build sam2...")
 
-    print("Attempting to download and build sam2...")
+    # print("download sam")
+    # sam_dir = snapshot_download(repo_id="alexnasa/sam2")
 
-    print("download sam")
-    sam_dir = snapshot_download(repo_id="alexnasa/sam2")
+    # @spaces.GPU(duration=500)
+    # def install_sam():
 
-    @spaces.GPU(duration=500)
-    def install_sam():
+    #     os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
+    #     sh(f"cd {sam_dir} && python setup.py build_ext --inplace && pip install -e .")
 
-        os.environ["TORCH_CUDA_ARCH_LIST"] = "9.0"
-        sh(f"cd {sam_dir} && python setup.py build_ext --inplace && pip install -e .")
+    # print("install sam")
+    # install_sam()
+    
+    print("Attempting to download")
 
-    print("install sam")
-    install_sam()
+    print("download sam")
+    snapshot_download(repo_id="alexnasa/sam2_C", local_dir=f"{os.getcwd()}" )
 
     # tell Python to re-scan site-packages now that the egg-link exists
     import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()

wan/modules/animate/preprocess/process_pipepline.py

@@ -20,7 +20,7 @@ from human_visualization import draw_aapose_by_meta_new
 from retarget_pose import get_retarget_pose
 import sam2.modeling.sam.transformer as transformer
 transformer.USE_FLASH_ATTN = True
-transformer.MATH_KERNEL_ON = True
+transformer.MATH_KERNEL_ON = False
 transformer.OLD_GPU = False
 from sam_utils import build_sam2_video_predictor