Skip to content

Commit 3770571

Browse files
authored
[tests] Fix Qwen test_inference slices (#12070)
update
1 parent 5937e11 commit 3770571

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/pipelines/qwenimage/test_qwenimage.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def test_inference(self):
160160
self.assertEqual(generated_image.shape, (3, 32, 32))
161161

162162
# fmt: off
163-
expected_slice = torch.tensor([0.563, 0.6358, 0.6028, 0.5656, 0.5806, 0.5512, 0.5712, 0.6331, 0.4147, 0.3558, 0.5625, 0.4831, 0.4957, 0.5258, 0.4075, 0.5018])
163+
expected_slice = torch.tensor([0.56331, 0.63677, 0.6015, 0.56369, 0.58166, 0.55277, 0.57176, 0.63261, 0.41466, 0.35561, 0.56229, 0.48334, 0.49714, 0.52622, 0.40872, 0.50208])
164164
# fmt: on
165165

166166
generated_slice = generated_image.flatten()

0 commit comments

Comments
 (0)