Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
fefed445
Unverified
Commit
fefed445
authored
Jan 15, 2024
by
YiYi Xu
Committed by
GitHub
Jan 15, 2024
Browse files
update slow test for SDXL k-diffusion pipeline (#6588)
update expected slice
parent
814f56d2
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_k_diffusion.py
...able_diffusion_xl/test_stable_diffusion_xl_k_diffusion.py
+2
-2
No files found.
tests/pipelines/stable_diffusion_xl/test_stable_diffusion_xl_k_diffusion.py
View file @
fefed445
...
...
@@ -64,7 +64,7 @@ class StableDiffusionXLKPipelineIntegrationTests(unittest.TestCase):
assert
image
.
shape
==
(
1
,
512
,
512
,
3
)
expected_slice
=
np
.
array
(
[
0.79
80453
4
,
0.79
81539
,
0.80
19961
,
0.7936565
,
0.7892033
,
0.79
1471
3
,
0.
7792827
,
0.77754563
,
0.7836
789
]
[
0.79
60002
4
,
0.79
6546
,
0.80
682373
,
0.79428387
,
0.79
0574
3
,
0.
8008807
,
0.786183
,
0.7835959
,
0.79
789
2
]
)
assert
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
...
...
@@ -97,7 +97,7 @@ class StableDiffusionXLKPipelineIntegrationTests(unittest.TestCase):
assert
image
.
shape
==
(
1
,
512
,
512
,
3
)
expected_slice
=
np
.
array
(
[
0.9
704869
,
0.9714559
,
0.9693254
,
0.96892524
,
0.9685236
,
0.9659081
,
0.9666761
,
0.9619067
,
0.961759
]
[
0.9
389532
,
0.9408587
,
0.9394901
,
0.939082
,
0.9402114
,
0.9382007
,
0.93737566
,
0.9346897
,
0.9324472
]
)
assert
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment