Skip to content

GitLab

Unverified Commit b5726358 authored by Sayak Paul's avatar Sayak Paul Committed by GitHub
Browse files

[Tests] add slow and nightly markers to sd3 lora integation. (#10458) 

add slow and nightly markers to sd3 lora integation.
parent fdcbbdf0
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
tests/lora/test_lora_layers_sd3.py
View file @ b5726358
...@@ -29,9 +29,11 @@ from diffusers import ( ...@@ -29,9 +29,11 @@ from diffusers import (
from diffusers.utils import load_image from diffusers.utils import load_image
from diffusers.utils.import_utils import is_accelerate_available from diffusers.utils.import_utils import is_accelerate_available
from diffusers.utils.testing_utils import ( from diffusers.utils.testing_utils import (
nightly,
numpy_cosine_similarity_distance, numpy_cosine_similarity_distance,
require_peft_backend, require_peft_backend,
require_torch_gpu, require_torch_gpu,
slow,
torch_device, torch_device,
) )
...@@ -126,6 +128,8 @@ class SD3LoRATests(unittest.TestCase, PeftLoraLoaderMixinTests): ...@@ -126,6 +128,8 @@ class SD3LoRATests(unittest.TestCase, PeftLoraLoaderMixinTests):
pass pass
@slow
@nightly
@require_torch_gpu @require_torch_gpu
@require_peft_backend @require_peft_backend
class LoraSD3IntegrationTests(unittest.TestCase): class LoraSD3IntegrationTests(unittest.TestCase):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment