Skip to content

GitLab

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

fix: lora sdxl tests (#4652)

parent d0c30cfd
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/models/test_lora_layers.py
View file @ a10107f9
......@@ -664,6 +664,7 @@ class SDXLLoraLoaderMixinTests(unittest.TestCase):
unet_lora_layers=lora_components["unet_lora_layers"],
text_encoder_lora_layers=lora_components["text_encoder_one_lora_layers"],
text_encoder_2_lora_layers=lora_components["text_encoder_two_lora_layers"],
safe_serialization=False,
)
self.assertTrue(os.path.isfile(os.path.join(tmpdirname, "pytorch_lora_weights.bin")))
sd_pipe.load_lora_weights(os.path.join(tmpdirname, "pytorch_lora_weights.bin"))
......
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