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
31010ecc
Unverified
Commit
31010ecc
authored
Oct 07, 2024
by
Sayak Paul
Committed by
GitHub
Oct 07, 2024
Browse files
[Chore] add a note on the versions in Flux LoRA integration tests (#9598)
add a note on the versions.
parent
3159e60d
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
5 additions
and
1 deletion
+5
-1
tests/lora/test_lora_layers_flux.py
tests/lora/test_lora_layers_flux.py
+5
-1
No files found.
tests/lora/test_lora_layers_flux.py
View file @
31010ecc
...
@@ -169,7 +169,11 @@ class FluxLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
...
@@ -169,7 +169,11 @@ class FluxLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
@
unittest
.
skip
(
"We cannot run inference on this model with the current CI hardware"
)
@
unittest
.
skip
(
"We cannot run inference on this model with the current CI hardware"
)
# TODO (DN6, sayakpaul): move these tests to a beefier GPU
# TODO (DN6, sayakpaul): move these tests to a beefier GPU
class
FluxLoRAIntegrationTests
(
unittest
.
TestCase
):
class
FluxLoRAIntegrationTests
(
unittest
.
TestCase
):
"""internal note: The integration slices were obtained on audace."""
"""internal note: The integration slices were obtained on audace.
torch: 2.6.0.dev20241006+cu124 with CUDA 12.5. Need the same setup for the
assertions to pass.
"""
num_inference_steps
=
10
num_inference_steps
=
10
seed
=
0
seed
=
0
...
...
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