PengWeixuanSZU commited on
Commit
d24619f
·
verified ·
1 Parent(s): f8d7756

Update pipeline_minimax_remover.py

Browse files
Files changed (1) hide show
  1. pipeline_minimax_remover.py +1 -0
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