Skip to content

GitLab

Unverified Commit fe574c8b authored by Dhruv Nair's avatar Dhruv Nair Committed by GitHub
Browse files

LoRA Unfusion test fix (#6291) 



update
Co-authored-by: default avatarSayak Paul <spsayakpaul@gmail.com>
parent 90b94799
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
tests/lora/test_lora_layers_peft.py
View file @ fe574c8b
...@@ -1881,7 +1881,9 @@ class LoraSDXLIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase): ...@@ -1881,7 +1881,9 @@ class LoraSDXLIntegrationTests(PeftLoraLoaderMixinTests, unittest.TestCase):
).images ).images
images_without_fusion = images.flatten() images_without_fusion = images.flatten()
self.assertTrue(np.allclose(images_with_fusion, images_without_fusion, atol=1e-3)) max_diff = numpy_cosine_similarity_distance(images_with_fusion, images_without_fusion)
assert max_diff < 1e-4
release_memory(pipe) release_memory(pipe)
def test_sdxl_1_0_lora_unfusion_effectivity(self): def test_sdxl_1_0_lora_unfusion_effectivity(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment