Skip to content

Commit 12fec91

Browse files
committed
[Chore] add a note on the versions in Flux LoRA integration tests (#9598)
add a note on the versions.
1 parent 4528d4a commit 12fec91

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

tests/lora/test_lora_layers_flux.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,11 @@ def test_modify_padding_mode(self):
169169
@unittest.skip("We cannot run inference on this model with the current CI hardware")
170170
# TODO (DN6, sayakpaul): move these tests to a beefier GPU
171171
class FluxLoRAIntegrationTests(unittest.TestCase):
172-
"""internal note: The integration slices were obtained on audace."""
172+
"""internal note: The integration slices were obtained on audace.
173+
174+
torch: 2.6.0.dev20241006+cu124 with CUDA 12.5. Need the same setup for the
175+
assertions to pass.
176+
"""
173177

174178
num_inference_steps = 10
175179
seed = 0

0 commit comments

Comments
 (0)