Spaces:

XiaRho
/

SEMat

Running on Zero

XiaRho commited on 6 days ago

Commit

3e2f8ca

•

1 Parent(s): 4cfe544

Update configs/SEMat_SAM2.py

Files changed (1) hide show

configs/SEMat_SAM2.py CHANGED Viewed

@@ -30,7 +30,7 @@ model.sam2 = True
 model.sam_model = L(build_sam2)(
     config_file = 'sam2_hiera_l.yaml',
     ckpt_path = None,
-    device = "cuda",
     bbox_mask_matting_token = True,
     mode="train",
     upscaled_embedding_res_add = False

 model.sam_model = L(build_sam2)(
     config_file = 'sam2_hiera_l.yaml',
     ckpt_path = None,
+    device = "cpu",
     bbox_mask_matting_token = True,
     mode="train",
     upscaled_embedding_res_add = False