Skip to content

Commit fe574c8

Browse files
DN6sayakpaul
andauthored
LoRA Unfusion test fix (#6291)
update Co-authored-by: Sayak Paul <[email protected]>
1 parent 90b9479 commit fe574c8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/lora/test_lora_layers_peft.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1881,7 +1881,9 @@ def test_sdxl_1_0_lora_unfusion(self):
18811881
).images
18821882
images_without_fusion = images.flatten()
18831883

1884-
self.assertTrue(np.allclose(images_with_fusion, images_without_fusion, atol=1e-3))
1884+
max_diff = numpy_cosine_similarity_distance(images_with_fusion, images_without_fusion)
1885+
assert max_diff < 1e-4
1886+
18851887
release_memory(pipe)
18861888

18871889
def test_sdxl_1_0_lora_unfusion_effectivity(self):

0 commit comments

Comments
 (0)