Skip to content

Commit 2e07b03

Browse files
committed
Update test_lora_layers_sd3.py
1 parent fdf4290 commit 2e07b03

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tests/lora/test_lora_layers_sd3.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -172,9 +172,8 @@ def test_sd3_img2img_lora(self):
172172
inputs = self.get_inputs(torch_device)
173173

174174
image = pipe(**inputs).images[0]
175-
image_slice = image[0, :10, :-1]
176-
expected_slice = np.array([0.67041016, 0.69921875, 0.6640625, 0.66796875, 0.65234375, 0.6665039, 0.67578125, 0.67626953, 0.67871094, 0.6928711])
177-
175+
image_slice = image[0, -3:, -3:]
176+
expected_slice = np.array([0.5396, 0.5776, 0.7432, 0.5151, 0.5586, 0.7383, 0.5537, 0.5933, 0.7153])
178177
max_diff = numpy_cosine_similarity_distance(expected_slice.flatten(), image_slice.flatten())
179178

180179
assert max_diff < 1e-4, f"Outputs are not close enough, got {max_diff}"

0 commit comments

Comments
 (0)