From b5726358cf125f2fa1a596dce321e91a225a57e4 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Mon, 6 Jan 2025 07:29:04 +0530 Subject: [PATCH] [Tests] add slow and nightly markers to sd3 lora integation. (#10458) add slow and nightly markers to sd3 lora integation. --- tests/lora/test_lora_layers_sd3.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/lora/test_lora_layers_sd3.py b/tests/lora/test_lora_layers_sd3.py index 8c42f9c86e..40383e3f1e 100644 --- a/tests/lora/test_lora_layers_sd3.py +++ b/tests/lora/test_lora_layers_sd3.py @@ -29,9 +29,11 @@ from diffusers import ( from diffusers.utils import load_image from diffusers.utils.import_utils import is_accelerate_available from diffusers.utils.testing_utils import ( + nightly, numpy_cosine_similarity_distance, require_peft_backend, require_torch_gpu, + slow, torch_device, ) @@ -126,6 +128,8 @@ class SD3LoRATests(unittest.TestCase, PeftLoraLoaderMixinTests): pass +@slow +@nightly @require_torch_gpu @require_peft_backend class LoraSD3IntegrationTests(unittest.TestCase):