AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

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

Browse Source 
add a note on the versions.
This commit is contained in:
Sayak Paul
2024-10-07 17:43:48 +05:30
committed by GitHub
parent 3159e60d59
commit 31010ecc45
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/lora/test_lora_layers_flux.py
View File

@@ -169,7 +169,11 @@ class FluxLoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
@unittest.skip("We cannot run inference on this model with the current CI hardware")
# TODO (DN6, sayakpaul): move these tests to a beefier GPU
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
seed = 0