From e2773c6255dbe0f5794516d707451531e9842ffd Mon Sep 17 00:00:00 2001 From: YiYi Xu Date: Fri, 19 Jan 2024 17:37:29 -1000 Subject: [PATCH] fix SDXL-kdiffusion tests (#6647) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 🤞🤞🤞 Co-authored-by: yiyixuxu --- .../stable_diffusion_xl/test_stable_diffusion_xl_k_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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 index e9758bf6eb..da10a0659d 100644 --- 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 @@ class StableDiffusionXLKPipelineIntegrationTests(unittest.TestCase): 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