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
Hide 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):
).images
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)
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