"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "08c852290a060b8ac21f263f897e1c9e25e1eda3"
Unverified Commit 31010ecc authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Chore] add a note on the versions in Flux LoRA integration tests (#9598)

add a note on the versions.
parent 3159e60d
...@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment