Skip to content

GitLab

Unverified Commit e2773c62 authored by YiYi Xu's avatar YiYi Xu Committed by GitHub
Browse files

fix SDXL-kdiffusion tests (#6647) 

🤞🤞🤞

Co-authored-by: default avataryiyixuxu <yixu310@gmail,com>
parent ac61eefc
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_k_diffusion.py
View file @ e2773c62
...@@ -97,7 +97,7 @@ class StableDiffusionXLKPipelineIntegrationTests(unittest.TestCase): ...@@ -97,7 +97,7 @@ class StableDiffusionXLKPipelineIntegrationTests(unittest.TestCase):
assert image.shape == (1, 512, 512, 3) assert image.shape == (1, 512, 512, 3)
expected_slice = np.array( 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 assert np.abs(image_slice.flatten() - expected_slice).max() < 1e-2
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment