[LoRA] fix dora test to catch the warning properly. (#9627)

fix dora test.
huggingface · Oct 10, 2024 · e16fd93 · e16fd93
1 parent 07bd2fa
commit e16fd93
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,