MiniMax-Remover

Running on Zero

PengWeixuanSZU commited on 26 days ago

Commit

7277a9d

verified ·

1 Parent(s): 8c24d5c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -192,7 +192,7 @@ def inference_and_return_video(dilation_iterations, num_inference_steps, video_s
     print("mask_tensor shape:",mask_tensor.shape)
     img_tensor=img_tensor.to("cuda")
     mask_tensor=mask_tensor.to("cuda")
-    mask_tensor = mask_tensor[:,:,:]
     pipe.to("cuda")

     print("mask_tensor shape:",mask_tensor.shape)
     img_tensor=img_tensor.to("cuda")
     mask_tensor=mask_tensor.to("cuda")
+    mask_tensor = mask_tensor[:,:,:,:1]
     pipe.to("cuda")