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
9f06a0d1
Unverified
Commit
9f06a0d1
authored
Jan 10, 2025
by
Sayak Paul
Committed by
GitHub
Jan 10, 2025
Browse files
[CI] Match remaining assertions from big runner (#10521)
* print * remove print. * print * update slice. * empty
parent
52c05bd4
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
2 additions
and
2 deletions
+2
-2
tests/lora/test_lora_layers_sd3.py
tests/lora/test_lora_layers_sd3.py
+1
-1
tests/quantization/bnb/test_mixed_int8.py
tests/quantization/bnb/test_mixed_int8.py
+1
-1
No files found.
tests/lora/test_lora_layers_sd3.py
View file @
9f06a0d1
...
...
@@ -177,7 +177,7 @@ class SD3LoraIntegrationTests(unittest.TestCase):
image
=
pipe
(
**
inputs
).
images
[
0
]
image_slice
=
image
[
0
,
-
3
:,
-
3
:]
expected_slice
=
np
.
array
([
0.5
396
,
0.5
776
,
0.
7432
,
0.5
151
,
0.5
586
,
0.
738
3
,
0.5
5
37
,
0.5
933
,
0.
7153
])
expected_slice
=
np
.
array
([
0.5
649
,
0.5
405
,
0.
5488
,
0.5
688
,
0.5
449
,
0.
551
3
,
0.5
3
37
,
0.5
107
,
0.
5059
])
max_diff
=
numpy_cosine_similarity_distance
(
expected_slice
.
flatten
(),
image_slice
.
flatten
())
...
...
tests/quantization/bnb/test_mixed_int8.py
View file @
9f06a0d1
...
...
@@ -372,7 +372,7 @@ class SlowBnb8bitTests(Base8bitTests):
output_type
=
"np"
,
).
images
out_slice
=
output
[
0
,
-
3
:,
-
3
:,
-
1
].
flatten
()
expected_slice
=
np
.
array
([
0.0
376
,
0.0
359
,
0.0
015
,
0.0
449
,
0.0
479
,
0.0
098
,
0.0
083
,
0.0
295
,
0.0
295
])
expected_slice
=
np
.
array
([
0.0
674
,
0.0
623
,
0.0
364
,
0.0
632
,
0.0
671
,
0.0
430
,
0.0
317
,
0.0
493
,
0.0
583
])
max_diff
=
numpy_cosine_similarity_distance
(
expected_slice
,
out_slice
)
self
.
assertTrue
(
max_diff
<
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