From 31010ecc45324c9ee294eb34cf5ad3eaf2cb8c3f Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Mon, 7 Oct 2024 17:43:48 +0530 Subject: [PATCH] [Chore] add a note on the versions in Flux LoRA integration tests (#9598) add a note on the versions. --- tests/lora/test_lora_layers_flux.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/tests/lora/test_lora_layers_flux.py b/tests/lora/test_lora_layers_flux.py index a75f9df910..4629c24c8c 100644 --- a/tests/lora/test_lora_layers_flux.py +++ b/tests/lora/test_lora_layers_flux.py @@ -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