remove

Author: sayakpaul

tests/lora/test_lora_layers_flux.py

@@ -215,7 +215,6 @@ def test_flux_the_last_ben(self):
             generator=torch.manual_seed(self.seed),
         ).images
         out_slice = out[0, -3:, -3:, -1].flatten()
-        print_tensor_test(out_slice)
         expected_slice = np.array([0.1855, 0.1855, 0.1836, 0.1855, 0.1836, 0.1875, 0.1777, 0.1758, 0.2246])
 
         max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), out_slice)
@@ -241,7 +240,6 @@ def test_flux_kohya(self):
         ).images
 
         out_slice = out[0, -3:, -3:, -1].flatten()
-        print_tensor_test(out_slice)
         expected_slice = np.array([0.6367, 0.6367, 0.6328, 0.6367, 0.6328, 0.6289, 0.6367, 0.6328, 0.6484])
 
         max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), out_slice)
@@ -264,7 +262,6 @@ def test_flux_kohya_with_text_encoder(self):
         ).images
 
         out_slice = out[0, -3:, -3:, -1].flatten()
-        print_tensor_test(out_slice)
         expected_slice = np.array([0.4023, 0.4023, 0.4023, 0.3965, 0.3984, 0.3965, 0.3926, 0.3906, 0.4219])
 
         max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), out_slice)
@@ -287,7 +284,6 @@ def test_flux_xlabs(self):
             generator=torch.manual_seed(self.seed),
         ).images
         out_slice = out[0, -3:, -3:, -1].flatten()
-        print_tensor_test(out_slice)
         expected_slice = np.array([0.3965, 0.4180, 0.4434, 0.4082, 0.4375, 0.4590, 0.4141, 0.4375, 0.4980])
 
         max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), out_slice)

tests/lora/test_lora_layers_sd3.py

@@ -184,7 +184,7 @@ def test_sd3_img2img_lora(self):
 
         image = pipe(**inputs).images[0]
         image_slice = image[0, :10, :10]
-        print_tensor_test(image[0, -3:, -3:, -1].flatten())
+        print_tensor_test(image[0, :10, :10].flatten())
         expected_slice = np.array(
             [
                 0.47827148,