Spaces:

facebook
/

EdgeTAM

Running on Zero

chongzhou commited on 3 days ago

Commit

eafda84

1 Parent(s): e64f815

second try

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,9 +73,9 @@ OBJ_ID = 0
 sam2_checkpoint = "checkpoints/edgetam.pt"
 model_cfg = "edgetam.yaml"
-predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint, device="cuda")
-print("predictor loaded")
 predictor.to("cuda")
 # use bfloat16 for the entire demo
 torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()

 sam2_checkpoint = "checkpoints/edgetam.pt"
 model_cfg = "edgetam.yaml"
+predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint, device="cpu")
 predictor.to("cuda")
+print("predictor loaded")
 # use bfloat16 for the entire demo
 torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()