print

Author: sayakpaul

tests/lora/test_lora_layers_sd3.py

@@ -36,6 +36,8 @@
     require_peft_backend,
     require_torch_gpu,
     torch_device,
+    print_tensor_test
+    
 )
 
 
@@ -177,6 +179,7 @@ def test_sd3_img2img_lora(self):
 
         image = pipe(**inputs).images[0]
         image_slice = image[0, -3:, -3:]
+        print_tensor_test(image_slice)
         expected_slice = np.array([0.5396, 0.5776, 0.7432, 0.5151, 0.5586, 0.7383, 0.5537, 0.5933, 0.7153])
 
         max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), image_slice.flatten())