add log

src/multiview_consist_edit/infer_tryon_multi.py

@@ -131,7 +131,7 @@ def main():
 
     image_idx = 0
     for i, batch in enumerate(test_dataloader):
-
+        print(f'Processing batch {i+1}/{len(test_dataloader)}')
         pixel_values = batch["pixel_values"]
         pixel_values_pose = batch["pixel_values_pose"].to(device='cuda')
         pixel_values_agnostic = batch["pixel_values_agnostic"].to(device='cuda')
@@ -147,7 +147,7 @@ def main():
         multi_length = pixel_values.shape[1]
         # dino_fea = dino_fea.unsqueeze(1)
         # print(dino_fea.shape) # [bs,1,768]
-        print(img_name)
+        # print(img_name)
         edited_images = pipe(
             num_inference_steps=num_inference_steps,
             guidance_scale=guidance_scale,