Spaces:

merve
/

SAM3-video-segmentation

Running on Zero

hysts HF Staff commited on 14 days ago

Commit

b350359

1 Parent(s): 0551705

Clean up

Files changed (1) hide show

app.py CHANGED Viewed

@@ -173,6 +173,7 @@ def init_video_session(
         GLOBAL_STATE.inference_session = processor.init_video_session(
             video=frames,
             inference_device=DEVICE,
             processing_device="cpu",
             video_storage_device="cpu",
             dtype=DTYPE,
@@ -182,9 +183,9 @@ def init_video_session(
         GLOBAL_STATE.inference_session = processor.init_video_session(
             video=raw_video,
             inference_device=DEVICE,
-            video_storage_device="cpu",
-            processing_device="cpu",
             inference_state_device=DEVICE,
             dtype=DTYPE,
         )

         GLOBAL_STATE.inference_session = processor.init_video_session(
             video=frames,
             inference_device=DEVICE,
+            inference_state_device=DEVICE,
             processing_device="cpu",
             video_storage_device="cpu",
             dtype=DTYPE,
         GLOBAL_STATE.inference_session = processor.init_video_session(
             video=raw_video,
             inference_device=DEVICE,
             inference_state_device=DEVICE,
+            processing_device="cpu",
+            video_storage_device="cpu",
             dtype=DTYPE,
         )