quick fix
Browse files
app.py
CHANGED
|
@@ -75,6 +75,7 @@ sam2_checkpoint = "checkpoints/edgetam.pt"
|
|
| 75 |
model_cfg = "edgetam.yaml"
|
| 76 |
predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint, device="cuda")
|
| 77 |
print("predictor loaded")
|
|
|
|
| 78 |
|
| 79 |
# use bfloat16 for the entire demo
|
| 80 |
torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
|
|
|
|
| 75 |
model_cfg = "edgetam.yaml"
|
| 76 |
predictor = build_sam2_video_predictor(model_cfg, sam2_checkpoint, device="cuda")
|
| 77 |
print("predictor loaded")
|
| 78 |
+
predictor.to("cuda")
|
| 79 |
|
| 80 |
# use bfloat16 for the entire demo
|
| 81 |
torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
|