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
effe4b97
"tests/vscode:/vscode.git/clone" did not exist on "a38bb5d1c6eef8ea6733b6925b3af187ce3fa5e4"
Unverified
Commit
effe4b97
authored
Jun 27, 2024
by
Dhruv Nair
Committed by
GitHub
Jun 26, 2024
Browse files
Update xformers SD3 test (#8712)
update
parent
5b51ad00
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
0 deletions
+8
-0
tests/models/test_modeling_common.py
tests/models/test_modeling_common.py
+4
-0
tests/pipelines/controlnet_sd3/test_controlnet_sd3.py
tests/pipelines/controlnet_sd3/test_controlnet_sd3.py
+4
-0
No files found.
tests/models/test_modeling_common.py
View file @
effe4b97
...
@@ -382,6 +382,10 @@ class ModelTesterMixin:
...
@@ -382,6 +382,10 @@ class ModelTesterMixin:
# If not has `set_attn_processor`, skip test
# If not has `set_attn_processor`, skip test
return
return
if
not
hasattr
(
model
,
"set_default_attn_processor"
):
# If not has `set_attn_processor`, skip test
return
model
.
set_default_attn_processor
()
model
.
set_default_attn_processor
()
assert
all
(
type
(
proc
)
==
AttnProcessor
for
proc
in
model
.
attn_processors
.
values
())
assert
all
(
type
(
proc
)
==
AttnProcessor
for
proc
in
model
.
attn_processors
.
values
())
with
torch
.
no_grad
():
with
torch
.
no_grad
():
...
...
tests/pipelines/controlnet_sd3/test_controlnet_sd3.py
View file @
effe4b97
...
@@ -190,6 +190,10 @@ class StableDiffusion3ControlNetPipelineFastTests(unittest.TestCase, PipelineTes
...
@@ -190,6 +190,10 @@ class StableDiffusion3ControlNetPipelineFastTests(unittest.TestCase, PipelineTes
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
np
.
abs
(
image_slice
.
flatten
()
-
expected_slice
).
max
()
<
1e-2
),
f
"Expected:
{
expected_slice
}
, got:
{
image_slice
.
flatten
()
}
"
),
f
"Expected:
{
expected_slice
}
, got:
{
image_slice
.
flatten
()
}
"
@
unittest
.
skip
(
"xFormersAttnProcessor does not work with SD3 Joint Attention"
)
def
test_xformers_attention_forwardGenerator_pass
(
self
):
pass
@
slow
@
slow
@
require_torch_gpu
@
require_torch_gpu
...
...
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