Update unidiffuser/sample_v1_test.py

Revert printing out expected image slices

unidiffuser/sample_v1_test.py

@@ -653,10 +653,10 @@ def evaluate(config):
 
         logging.debug(f"Generated text: {prompts}")
 
-        numpy_sample = samples[0].clone().cpu().permute(1, 2, 0).float().numpy()
-        numpy_sample_slice = numpy_sample[-3:, -3:, -1].flatten()
-        print(f"Expected slice: {numpy_sample_slice}")
-        print(f"Expected text: {prompts}")
+        # numpy_sample = samples[0].clone().cpu().permute(1, 2, 0).float().numpy()
+        # numpy_sample_slice = numpy_sample[-3:, -3:, -1].flatten()
+        # print(f"Expected slice: {numpy_sample_slice}")
+        # print(f"Expected text: {prompts}")
 
         output_images = samples
         output_text = prompts
@@ -672,9 +672,9 @@ def evaluate(config):
             _text = sample_fn('i2t', z=z_img, clip_img=clip_imgs)  # conditioned on the image embedding
             _z, _clip_img = sample_fn('t2i', text=_text)
         samples = unpreprocess(decode(_z))
-        numpy_sample = samples[0].clone().cpu().permute(1, 2, 0).float().numpy()
-        numpy_sample_slice = numpy_sample[-3:, -3:, -1].flatten()
-        print(f"Expected slice: {numpy_sample_slice}")
+        # numpy_sample = samples[0].clone().cpu().permute(1, 2, 0).float().numpy()
+        # numpy_sample_slice = numpy_sample[-3:, -3:, -1].flatten()
+        # print(f"Expected slice: {numpy_sample_slice}")
         output_images = samples