Update model.py

model.py

@@ -1138,7 +1138,6 @@ class StreamMultiDiffusion(nn.Module):
                 bootstrap_mask_ = torch.concat([bootstrap_mask, bootstrap_mask], dim=0)
             else:
                 bootstrap_mask_ = bootstrap_mask
-            print('2222222222222222222222222222222222222', model_pred.shape, bootstrap_mask_)
             model_pred = shift_to_mask_bbox_center(model_pred, bootstrap_mask_)
             x_t_latent = shift_to_mask_bbox_center(x_t_latent, bootstrap_mask)
 
@@ -1223,7 +1222,6 @@ class StreamMultiDiffusion(nn.Module):
             self.stock_noise_ = self.stock_noise.repeat_interleave(self.num_layers, dim=0)  # (T * p, 77, 768)
 
         x_0_pred_batch = self.unet_step(latent)
-        print('111111111111111111111111111111111')
         latent = x_0_pred_batch[-1:]
         self.x_t_latent_buffer = (
             self.alpha_prod_t_sqrt[1:] * x_0_pred_batch[:-1]