update

huggingface · Dec 22, 2023 · a988009 · a988009
1 parent 3369bc8
commit a988009
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/lora/test_lora_layers_peft.py b/tests/lora/test_lora_layers_peft.py
@@ -1877,7 +1877,9 @@ def test_sdxl_1_0_lora_unfusion(self):
         ).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):