From 194f074e52927d8b9c06cf03aca654d4bdaee317 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Thu, 17 Aug 2023 14:29:41 +0530 Subject: [PATCH] fix: lora sdxl tests --- tests/models/test_lora_layers.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/models/test_lora_layers.py b/tests/models/test_lora_layers.py index 7f06da81ba38..c2fe98993d00 100644 --- a/tests/models/test_lora_layers.py +++ b/tests/models/test_lora_layers.py @@ -664,6 +664,7 @@ def test_load_lora_locally(self): 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"))