Skip to content

Commit 1b71c5c

Browse files
committed
print test
1 parent 9d3d68f commit 1b71c5c

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/lora/test_lora_layers_sdxl.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -628,6 +628,8 @@ def test_integration_logits_for_dora_lora(self):
628628
).images
629629

630630
predicted_slice = images[0, -3:, -3:, -1].flatten()
631+
from diffusers.utils.testing_utils import print_tensor_test
632+
print_tensor_test(predicted_slice)
631633
expected_slice_scale = np.array([0.3932, 0.3742, 0.4429, 0.3737, 0.3504, 0.433, 0.3948, 0.3769, 0.4516])
632634
max_diff = numpy_cosine_similarity_distance(expected_slice_scale, predicted_slice)
633635
assert max_diff < 1e-3

0 commit comments

Comments
 (0)