Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline_minimax_remover.py
Browse files
pipeline_minimax_remover.py
CHANGED
@@ -151,6 +151,7 @@ class Minimax_Remover_Pipeline(DiffusionPipeline):
|
|
151 |
masks[masks>0] = 1
|
152 |
images = rearrange(images, "f h w c -> c f h w")
|
153 |
images = self.resize(images[None,...], height, width).to("cuda:0").half()
|
|
|
154 |
|
155 |
masked_images = images * (1-masks)
|
156 |
|
|
|
151 |
masks[masks>0] = 1
|
152 |
images = rearrange(images, "f h w c -> c f h w")
|
153 |
images = self.resize(images[None,...], height, width).to("cuda:0").half()
|
154 |
+
print(f"images.shape: {images.shape}, masks.shape:{masks.shape}")
|
155 |
|
156 |
masked_images = images * (1-masks)
|
157 |
|