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
Hide 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): ...@@ -664,6 +664,7 @@ class SDXLLoraLoaderMixinTests(unittest.TestCase):
unet_lora_layers=lora_components["unet_lora_layers"], unet_lora_layers=lora_components["unet_lora_layers"],
text_encoder_lora_layers=lora_components["text_encoder_one_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"], 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"))) 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")) 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