Update sam2/build_sam.py
Browse files- sam2/build_sam.py +1 -1
sam2/build_sam.py
CHANGED
|
@@ -15,7 +15,7 @@ from omegaconf import OmegaConf
|
|
| 15 |
def build_sam2(
|
| 16 |
config_file,
|
| 17 |
ckpt_path=None,
|
| 18 |
-
device="
|
| 19 |
mode="eval",
|
| 20 |
hydra_overrides_extra=[],
|
| 21 |
apply_postprocessing=True,
|
|
|
|
| 15 |
def build_sam2(
|
| 16 |
config_file,
|
| 17 |
ckpt_path=None,
|
| 18 |
+
device="cpu",
|
| 19 |
mode="eval",
|
| 20 |
hydra_overrides_extra=[],
|
| 21 |
apply_postprocessing=True,
|