From e16fd93d0a40156c1f49fde07f6f2eb438983927 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Thu, 10 Oct 2024 11:47:49 +0530 Subject: [PATCH] [LoRA] fix dora test to catch the warning properly. (#9627) fix dora test. --- tests/lora/test_lora_layers_sdxl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/lora/test_lora_layers_sdxl.py b/tests/lora/test_lora_layers_sdxl.py index 8deecd770c31..94a44ed8f9ec 100644 --- a/tests/lora/test_lora_layers_sdxl.py +++ b/tests/lora/test_lora_layers_sdxl.py @@ -619,12 +619,12 @@ def test_integration_logits_multi_adapter(self): @nightly def test_integration_logits_for_dora_lora(self): pipeline = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0") - pipeline.load_lora_weights("hf-internal-testing/dora-trained-on-kohya") - pipeline.enable_model_cpu_offload() logger = logging.get_logger("diffusers.loaders.lora_pipeline") logger.setLevel(30) with CaptureLogger(logger) as cap_logger: + pipeline.load_lora_weights("hf-internal-testing/dora-trained-on-kohya") + pipeline.enable_model_cpu_offload() images = pipeline( "photo of ohwx dog", num_inference_steps=10,