Spaces:

PengWeixuanSZU
/

MiniMax-Remover

Running on Zero

PengWeixuanSZU commited on 25 days ago

Commit

6dd3fdf

verified ·

1 Parent(s): 2c6d33e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -246,7 +246,7 @@ def track_video(n_frames,video_state):
     global video_predictor
     video_predictor=video_predictor.to("cuda")
-    inference_state = video_predictor.init_state(images=images/255, device=device)
     video_state["inference_state"] = inference_state
     if len(torch.from_numpy(video_state["masks"][0]).shape) == 3:

     global video_predictor
     video_predictor=video_predictor.to("cuda")
+    inference_state = video_predictor.init_state(images=images/255, device="cuda")
     video_state["inference_state"] = inference_state
     if len(torch.from_numpy(video_state["masks"][0]).shape) == 3: