Merge branch 'main' into handle-missing-keys-lora

huggingface · Oct 10, 2024 · 1ba04a4 · 1ba04a4
2 parents 56cb020 + e16fd93
commit 1ba04a4
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,