fix: lora sdxl tests (huggingface#4652)

sayakpaul · web-flow · commit a10107f92b09 · 2023-08-17T15:59:50.000+05:30
diff --git 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"))

Original file line number	Diff line number	Diff line change
`@@ -664,6 +664,7 @@ def test_load_lora_locally(self):`
`664`	`664`	`unet_lora_layers=lora_components["unet_lora_layers"],`
`665`	`665`	`text_encoder_lora_layers=lora_components["text_encoder_one_lora_layers"],`
`666`	`666`	`text_encoder_2_lora_layers=lora_components["text_encoder_two_lora_layers"],`
	`667`	`+ safe_serialization=False,`
`667`	`668`	`)`
`668`	`669`	`self.assertTrue(os.path.isfile(os.path.join(tmpdirname, "pytorch_lora_weights.bin")))`
`669`	`670`	`sd_pipe.load_lora_weights(os.path.join(tmpdirname, "pytorch_lora_weights.bin"))`