Skip to content

Commit 763fd3b

Browse files
committed
mark wanvace test as flaky
1 parent 0aa92e8 commit 763fd3b

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tests/lora/test_lora_layers_wanvace.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from diffusers.utils.import_utils import is_peft_available
2929
from diffusers.utils.testing_utils import (
3030
floats_tensor,
31+
is_flaky,
3132
require_peft_backend,
3233
require_peft_version_greater,
3334
skip_mps,
@@ -215,3 +216,7 @@ def test_lora_exclude_modules_wanvace(self):
215216
np.allclose(output_lora_exclude_modules, output_lora_pretrained, atol=1e-3, rtol=1e-3),
216217
"Lora outputs should match.",
217218
)
219+
220+
@is_flaky
221+
def test_simple_inference_with_text_denoiser_lora_and_scale(self):
222+
super().test_simple_inference_with_text_denoiser_lora_and_scale()

0 commit comments

Comments
 (0)