Skip to content

GitLab

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

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

fix dora test.
parent 07bd2fab
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/lora/test_lora_layers_sdxl.py
View file @ e16fd93d
...@@ -619,12 +619,12 @@ class LoraSDXLIntegrationTests(unittest.TestCase): ...@@ -619,12 +619,12 @@ class LoraSDXLIntegrationTests(unittest.TestCase):
@nightly @nightly
def test_integration_logits_for_dora_lora(self): def test_integration_logits_for_dora_lora(self):
pipeline = StableDiffusionXLPipeline.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0") 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 = logging.get_logger("diffusers.loaders.lora_pipeline")
logger.setLevel(30) logger.setLevel(30)
with CaptureLogger(logger) as cap_logger: with CaptureLogger(logger) as cap_logger:
pipeline.load_lora_weights("hf-internal-testing/dora-trained-on-kohya")
pipeline.enable_model_cpu_offload()
images = pipeline( images = pipeline(
"photo of ohwx dog", "photo of ohwx dog",
num_inference_steps=10, num_inference_steps=10,
......
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