MiniMax-Remover

Running on Zero

PengWeixuanSZU commited on 25 days ago

Commit

886e420

verified ·

1 Parent(s): 6047cd6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -248,7 +248,7 @@ def track_video(n_frames,video_state):
     video_predictor_local = build_sam2_video_predictor(config, sam2_checkpoint, device="cuda")
     inference_state = video_predictor_local.init_state(images=images/255, device="cuda")
-    video_state["inference_state"] = inference_state
     if len(torch.from_numpy(video_state["masks"][0]).shape) == 3:
         mask = torch.from_numpy(video_state["masks"][0])[:,:,0]

     video_predictor_local = build_sam2_video_predictor(config, sam2_checkpoint, device="cuda")
     inference_state = video_predictor_local.init_state(images=images/255, device="cuda")
+    #video_state["inference_state"] = inference_state #cause bug
     if len(torch.from_numpy(video_state["masks"][0]).shape) == 3:
         mask = torch.from_numpy(video_state["masks"][0])[:,:,0]