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
49b7ccfb
Unverified
Commit
49b7ccfb
authored
May 18, 2023
by
Will Berman
Committed by
GitHub
May 18, 2023
Browse files
parameterize pass single args through tuple (#3477)
parent
7200985e
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
tests/models/test_models_vae.py
tests/models/test_models_vae.py
+2
-2
No files found.
tests/models/test_models_vae.py
View file @
49b7ccfb
...
...
@@ -321,7 +321,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase):
assert
torch_all_close
(
output_slice
,
expected_output_slice
,
atol
=
5e-3
)
@
parameterized
.
expand
([
13
,
16
,
27
])
@
parameterized
.
expand
([
(
13
,
),
(
16
,
),
(
27
,)
])
@
require_torch_gpu
@
unittest
.
skipIf
(
not
is_xformers_available
(),
reason
=
"xformers is not required when using PyTorch 2.0."
)
def
test_stable_diffusion_decode_xformers_vs_2_0_fp16
(
self
,
seed
):
...
...
@@ -339,7 +339,7 @@ class AutoencoderKLIntegrationTests(unittest.TestCase):
assert
torch_all_close
(
sample
,
sample_2
,
atol
=
1e-1
)
@
parameterized
.
expand
([
13
,
16
,
37
])
@
parameterized
.
expand
([
(
13
,
),
(
16
,
),
(
37
,)
])
@
require_torch_gpu
@
unittest
.
skipIf
(
not
is_xformers_available
(),
reason
=
"xformers is not required when using PyTorch 2.0."
)
def
test_stable_diffusion_decode_xformers_vs_2_0
(
self
,
seed
):
...
...
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