AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-27 17:22:53 +03:00
Code Activity

fix: lora sdxl tests (#4652)

Browse Source
This commit is contained in:
Sayak Paul
2023-08-17 15:59:50 +05:30
committed by GitHub
parent d0c30cfd37
commit a10107f92b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/models/test_lora_layers.py
View File

@@ -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"))