Skip to content

Commit f823914

Browse files
committed
Update new tests to coordinate with huggingface#2953
1 parent a6ffbd1 commit f823914

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/pipelines/stable_diffusion/test_stable_diffusion_pix2pix_zero.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -190,7 +190,7 @@ def test_stable_diffusion_pix2pix_zero_inversion(self):
190190
image = sd_pipe.invert(**inputs).images
191191
image_slice = image[0, -3:, -3:, -1]
192192
assert image.shape == (1, 32, 32, 3)
193-
expected_slice = np.array([0.4884, 0.4707, 0.5653, 0.4997, 0.4567, 0.4955, 0.5102, 0.5195, 0.515])
193+
expected_slice = np.array([0.4833, 0.4696, 0.5574, 0.5194, 0.5248, 0.5638, 0.5040, 0.5423, 0.5072])
194194

195195
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-3
196196

@@ -205,7 +205,7 @@ def test_stable_diffusion_pix2pix_zero_inversion_batch(self):
205205
image = sd_pipe.invert(**inputs).images
206206
image_slice = image[1, -3:, -3:, -1]
207207
assert image.shape == (2, 32, 32, 3)
208-
expected_slice = np.array([0.4747, 0.4161, 0.5004, 0.3475, 0.4258, 0.5463, 0.4562, 0.4967, 0.5166])
208+
expected_slice = np.array([0.6672, 0.5203, 0.4908, 0.4376, 0.4517, 0.5544, 0.4605, 0.4826, 0.5007])
209209

210210
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-3
211211

0 commit comments

Comments
 (0)