Skip to content

Commit 5a7997b

Browse files
committed
absent->extra
1 parent 08ea124 commit 5a7997b

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/lora/test_lora_layers_flux.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -251,14 +251,14 @@ def test_lora_expansion_works_for_extra_keys(self):
251251

252252
pipe.set_adapters(["one", "two"])
253253
self.assertTrue(check_if_lora_correctly_set(pipe.transformer), "Lora not correctly set in transformer")
254-
images_lora_with_absent_keys = pipe(**inputs, generator=torch.manual_seed(0)).images
254+
images_lora_with_extra_keys = pipe(**inputs, generator=torch.manual_seed(0)).images
255255

256256
self.assertFalse(
257-
np.allclose(images_lora, images_lora_with_absent_keys, atol=1e-3, rtol=1e-3),
257+
np.allclose(images_lora, images_lora_with_extra_keys, atol=1e-3, rtol=1e-3),
258258
"Different LoRAs should lead to different results.",
259259
)
260260
self.assertFalse(
261-
np.allclose(output_no_lora, images_lora_with_absent_keys, atol=1e-3, rtol=1e-3),
261+
np.allclose(output_no_lora, images_lora_with_extra_keys, atol=1e-3, rtol=1e-3),
262262
"LoRA should lead to different results.",
263263
)
264264

0 commit comments

Comments
 (0)