Skip to content

Commit ca9a7b8

Browse files
committed
Update new tests to coordinate with huggingface#2953
1 parent 135c390 commit ca9a7b8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/pipelines/stable_diffusion_2/test_stable_diffusion_diffedit.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -228,7 +228,7 @@ def test_mask(self):
228228
expected_slice = np.array([0] * 9)
229229
max_diff = np.abs(mask_slice.flatten() - expected_slice).max()
230230
self.assertLessEqual(max_diff, 1e-3)
231-
self.assertEqual(mask[0, -3, -4], 1)
231+
self.assertEqual(mask[0, -3, -4], 0)
232232

233233
def test_inversion(self):
234234
device = "cpu"
@@ -244,7 +244,7 @@ def test_inversion(self):
244244

245245
self.assertEqual(image.shape, (2, 32, 32, 3))
246246
expected_slice = np.array(
247-
[0.5588859, 0.535619, 0.52224344, 0.55604255, 0.48608556, 0.51105076, 0.50301707, 0.44348782, 0.48488846],
247+
[0.5150 ,0.5134 ,0.5043 ,0.5376 ,0.4694 ,0.51050 ,0.5015 ,0.4407 ,0.4799],
248248
)
249249
max_diff = np.abs(image_slice.flatten() - expected_slice).max()
250250
self.assertLessEqual(max_diff, 1e-3)

0 commit comments

Comments
 (0)