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

up

Browse Source
This commit is contained in:
sayakpaul
2025-10-03 21:00:05 +05:30
parent 4f5e9a665e
commit 0d3da485a0
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/lora/test_lora_layers_cogvideox.py
View File

@@ -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(

4
tests/lora/test_lora_layers_sdxl.py
View File

@@ -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
)