Revert fix for batch size 1

2024-10-01 21:50:42 +05:00
parent 03ff6c2da0
commit 09bb7452d9
1 changed files with 0 additions and 3 deletions
@@ -204,9 +204,6 @@ class RembgByBiRefNet:
        out_images = torch.cat(_images, dim=0)
        out_masks = torch.cat(_masks, dim=0)

-        if out_masks.shape[0] == 1:
-            out_masks = out_masks.squeeze(0)
-
        return out_images, out_masks