Spaces:

yansong1616
/

3dilize_anything

Sleeping

yansong1616 commited on Sep 6, 2024

Commit

e4d53e2

verified ·

1 Parent(s): fbb1647

Update SAM2/sam2/build_sam.py

Files changed (1) hide show

SAM2/sam2/build_sam.py CHANGED Viewed

@@ -43,7 +43,7 @@ def build_sam2(
 def build_sam2_video_predictor(
     config_file,
     ckpt_path=None,
-    device="cuda",
     mode="eval",
     hydra_overrides_extra=[],
     apply_postprocessing=True,

 def build_sam2_video_predictor(
     config_file,
     ckpt_path=None,
+    device="cuda" if torch.cuda.is_available() else "cpu",
     mode="eval",
     hydra_overrides_extra=[],
     apply_postprocessing=True,