@@ -289,7 +289,7 @@ def test_stable_diffusion_depth2img_default_case(self):
289
289
# expected_slice = np.array([0.6071, 0.5035, 0.4378, 0.5776, 0.5753, 0.4316, 0.4513, 0.5263, 0.4546])
290
290
# else:
291
291
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 ])
293
293
294
294
assert np .abs (image_slice .flatten () - expected_slice ).max () < 1e-3
295
295
@@ -311,7 +311,7 @@ def test_stable_diffusion_depth2img_negative_prompt(self):
311
311
# expected_slice = np.array([0.5825, 0.5135, 0.4095, 0.5452, 0.6059, 0.4211, 0.3994, 0.5177, 0.4335])
312
312
# else:
313
313
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 ])
315
315
316
316
assert np .abs (image_slice .flatten () - expected_slice ).max () < 1e-3
317
317
@@ -334,7 +334,7 @@ def test_stable_diffusion_depth2img_multiple_init_images(self):
334
334
# expected_slice = np.array([0.6501, 0.5150, 0.4939, 0.6688, 0.5437, 0.5758, 0.5115, 0.4406, 0.4551])
335
335
# else:
336
336
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 ])
338
338
339
339
assert np .abs (image_slice .flatten () - expected_slice ).max () < 1e-3
340
340
@@ -389,7 +389,7 @@ def test_stable_diffusion_depth2img_pil(self):
389
389
# if torch_device == "mps":
390
390
# expected_slice = np.array([0.53232, 0.47015, 0.40868, 0.45651, 0.4891, 0.4668, 0.4287, 0.48822, 0.47439])
391
391
# 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 ])
393
393
print_tensor_test (image_slice )
394
394
395
395
assert np .abs (image_slice .flatten () - expected_slice ).max () < 1e-3
0 commit comments