fix SDXL-kdiffusion tests (huggingface#6647)

yiyixuxu · yiyixuxu · web-flow · commit e2773c6255db · 2024-01-19T17:37:29.000-10:00
🤞🤞🤞

Co-authored-by: yiyixuxu &lt;yixu310@gmail,com&gt;
diff --git a/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_k_diffusion.py b/tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_k_diffusion.py
@@ -97,7 +97,7 @@ def test_stable_diffusion_karras_sigmas(self):
 
         assert image.shape == (1, 512, 512, 3)
         expected_slice = np.array(
-            [0.9389532, 0.9408587, 0.9394901, 0.939082, 0.9402114, 0.9382007, 0.93737566, 0.9346897, 0.9324472]
+            [0.9506951, 0.9527786, 0.95309967, 0.9511477, 0.952523, 0.9515326, 0.9511933, 0.9480397, 0.94930184]
         )
 
         assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,7 @@ def test_stable_diffusion_karras_sigmas(self):`
`97`	`97`
`98`	`98`	`assert image.shape == (1, 512, 512, 3)`
`99`	`99`	`expected_slice = np.array(`
`100`		`- [0.9389532, 0.9408587, 0.9394901, 0.939082, 0.9402114, 0.9382007, 0.93737566, 0.9346897, 0.9324472]`
	`100`	`+ [0.9506951, 0.9527786, 0.95309967, 0.9511477, 0.952523, 0.9515326, 0.9511933, 0.9480397, 0.94930184]`
`101`	`101`	`)`
`102`	`102`
`103`	`103`	`assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2`