Unverified Commit 9f06a0d1 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[CI] Match remaining assertions from big runner (#10521)

* print

* remove print.

* print

* update slice.

* empty
parent 52c05bd4
...@@ -177,7 +177,7 @@ class SD3LoraIntegrationTests(unittest.TestCase): ...@@ -177,7 +177,7 @@ class SD3LoraIntegrationTests(unittest.TestCase):
image = pipe(**inputs).images[0] image = pipe(**inputs).images[0]
image_slice = image[0, -3:, -3:] image_slice = image[0, -3:, -3:]
expected_slice = np.array([0.5396, 0.5776, 0.7432, 0.5151, 0.5586, 0.7383, 0.5537, 0.5933, 0.7153]) expected_slice = np.array([0.5649, 0.5405, 0.5488, 0.5688, 0.5449, 0.5513, 0.5337, 0.5107, 0.5059])
max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), image_slice.flatten()) max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), image_slice.flatten())
......
...@@ -372,7 +372,7 @@ class SlowBnb8bitTests(Base8bitTests): ...@@ -372,7 +372,7 @@ class SlowBnb8bitTests(Base8bitTests):
output_type="np", output_type="np",
).images ).images
out_slice = output[0, -3:, -3:, -1].flatten() out_slice = output[0, -3:, -3:, -1].flatten()
expected_slice = np.array([0.0376, 0.0359, 0.0015, 0.0449, 0.0479, 0.0098, 0.0083, 0.0295, 0.0295]) expected_slice = np.array([0.0674, 0.0623, 0.0364, 0.0632, 0.0671, 0.0430, 0.0317, 0.0493, 0.0583])
max_diff = numpy_cosine_similarity_distance(expected_slice, out_slice) max_diff = numpy_cosine_similarity_distance(expected_slice, out_slice)
self.assertTrue(max_diff < 1e-2) self.assertTrue(max_diff < 1e-2)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment