From 0d3da485a0ef247566113a2e368bbe6a9d550401 Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Fri, 3 Oct 2025 21:00:05 +0530 Subject: [PATCH] up --- tests/lora/test_lora_layers_cogvideox.py | 4 ++-- tests/lora/test_lora_layers_sdxl.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/lora/test_lora_layers_cogvideox.py b/tests/lora/test_lora_layers_cogvideox.py index f5d8564834..ad2943b681 100644 --- a/tests/lora/test_lora_layers_cogvideox.py +++ b/tests/lora/test_lora_layers_cogvideox.py @@ -124,9 +124,9 @@ class TestCogVideoXLoRA(PeftLoraLoaderMixinTests): def test_simple_inference_with_text_denoiser_lora_unfused(self, pipe): super().test_simple_inference_with_text_denoiser_lora_unfused(pipe=pipe, expected_atol=9e-3) - def test_lora_scale_kwargs_match_fusion(self, pipe, base_pipe_output): + def test_lora_scale_kwargs_match_fusion(self, base_pipe_output): super().test_lora_scale_kwargs_match_fusion( - pipe=pipe, base_pipe_output=base_pipe_output, expected_atol=9e-3, expected_rtol=9e-3 + base_pipe_output=base_pipe_output, expected_atol=9e-3, expected_rtol=9e-3 ) @pytest.mark.parametrize( diff --git a/tests/lora/test_lora_layers_sdxl.py b/tests/lora/test_lora_layers_sdxl.py index fd10d4eeda..e1bc6e8ecb 100644 --- a/tests/lora/test_lora_layers_sdxl.py +++ b/tests/lora/test_lora_layers_sdxl.py @@ -132,7 +132,7 @@ class TestStableDiffusionXLLoRA(PeftLoraLoaderMixinTests): pipe=pipe, expected_atol=expected_atol, expected_rtol=expected_rtol ) - def test_lora_scale_kwargs_match_fusion(self, base_pipe_output, pipe): + def test_lora_scale_kwargs_match_fusion(self, base_pipe_output): if torch.cuda.is_available(): expected_atol = 9e-2 expected_rtol = 9e-2 @@ -141,7 +141,7 @@ class TestStableDiffusionXLLoRA(PeftLoraLoaderMixinTests): expected_rtol = 1e-3 super().test_lora_scale_kwargs_match_fusion( - pipe=pipe, base_pipe_output=base_pipe_output, expected_atol=expected_atol, expected_rtol=expected_rtol + base_pipe_output=base_pipe_output, expected_atol=expected_atol, expected_rtol=expected_rtol )