Skip to content

Commit 64d1727

Browse files
correct all tests
1 parent ec6eb48 commit 64d1727

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

test_corrections.txt

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py;StableDiffusionDepth2ImgPipelineFastTests;test_stable_diffusion_depth2img_default_case;expected_slice = np.array([0.6312, 0.4984, 0.4154, 0.4788, 0.5535, 0.4599, 0.4017, 0.5359, 0.4716])
2+
tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py;StableDiffusionDepth2ImgPipelineFastTests;test_stable_diffusion_depth2img_multiple_init_images;expected_slice = np.array([0.6267, 0.5232, 0.6001, 0.6738, 0.5029, 0.6429, 0.5364, 0.4159, 0.4674])
3+
tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py;StableDiffusionDepth2ImgPipelineFastTests;test_stable_diffusion_depth2img_negative_prompt;expected_slice = np.array([0.6296, 0.5125, 0.3890, 0.4456, 0.5955, 0.4621, 0.3810, 0.5310, 0.4626])
4+
tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py;StableDiffusionDepth2ImgPipelineFastTests;test_stable_diffusion_depth2img_pil;expected_slice = np.array([0.6312, 0.4984, 0.4154, 0.4788, 0.5535, 0.4599, 0.4017, 0.5359, 0.4716])

tests/pipelines/stable_diffusion_2/test_stable_diffusion_depth.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -289,7 +289,7 @@ def test_stable_diffusion_depth2img_default_case(self):
289289
# expected_slice = np.array([0.6071, 0.5035, 0.4378, 0.5776, 0.5753, 0.4316, 0.4513, 0.5263, 0.4546])
290290
# else:
291291
print_tensor_test(image_slice)
292-
expected_slice = np.array([0.6374, 0.5039, 0.4199, 0.4819, 0.5563, 0.4617, 0.4028, 0.5381, 0.4711])
292+
expected_slice = np.array([0.6312, 0.4984, 0.4154, 0.4788, 0.5535, 0.4599, 0.4017, 0.5359, 0.4716])
293293

294294
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-3
295295

@@ -311,7 +311,7 @@ def test_stable_diffusion_depth2img_negative_prompt(self):
311311
# expected_slice = np.array([0.5825, 0.5135, 0.4095, 0.5452, 0.6059, 0.4211, 0.3994, 0.5177, 0.4335])
312312
# else:
313313
print_tensor_test(image_slice)
314-
expected_slice = np.array([0.6332, 0.5167, 0.3911, 0.4446, 0.5971, 0.4619, 0.3821, 0.5323, 0.4621])
314+
expected_slice = np.array([0.6296, 0.5125, 0.3890, 0.4456, 0.5955, 0.4621, 0.3810, 0.5310, 0.4626])
315315

316316
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-3
317317

@@ -334,7 +334,7 @@ def test_stable_diffusion_depth2img_multiple_init_images(self):
334334
# expected_slice = np.array([0.6501, 0.5150, 0.4939, 0.6688, 0.5437, 0.5758, 0.5115, 0.4406, 0.4551])
335335
# else:
336336
print_tensor_test(image_slice)
337-
expected_slice = np.array([0.6248, 0.5206, 0.6007, 0.6749, 0.5022, 0.6442, 0.5352, 0.4140, 0.4681])
337+
expected_slice = np.array([0.6267, 0.5232, 0.6001, 0.6738, 0.5029, 0.6429, 0.5364, 0.4159, 0.4674])
338338

339339
assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-3
340340

@@ -389,7 +389,7 @@ def test_stable_diffusion_depth2img_pil(self):
389389
# if torch_device == "mps":
390390
# expected_slice = np.array([0.53232, 0.47015, 0.40868, 0.45651, 0.4891, 0.4668, 0.4287, 0.48822, 0.47439])
391391
# else:
392-
expected_slice = np.array([0.6374, 0.5039, 0.4199, 0.4819, 0.5563, 0.4617, 0.4028, 0.5381, 0.4711])
392+
expected_slice = np.array([0.6312, 0.4984, 0.4154, 0.4788, 0.5535, 0.4599, 0.4017, 0.5359, 0.4716])
393393
print_tensor_test(image_slice)
394394

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

0 commit comments

Comments
 (0)