MiniMax-Remover

Running on Zero

PengWeixuanSZU commited on Jun 20

Commit

f8d7756

verified ·

1 Parent(s): c998093

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

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

     img_tensor, mask_tensor = preprocess_for_removal(images, masks)
     img_tensor=img_tensor.to("cuda")
     mask_tensor=mask_tensor.to("cuda")
     mask_tensor = mask_tensor[:,:,:]
     pipe.to("cuda")