fix float pointing error

huggingface · Jun 14, 2024 · fe07293 · fe07293
1 parent 8aae77b
commit fe07293
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ipex/test_modeling.py b/tests/ipex/test_modeling.py
@@ -446,7 +446,7 @@ def test_compare_to_transformers(self, model_arch):
         # Compare tensor outputs
         self.assertTrue(torch.allclose(outputs.logits, transformers_outputs.logits, atol=1e-4))
         self.assertTrue(torch.equal(outputs.logits, loaded_model_outputs.logits))
-        self.assertTrue(torch.equal(outputs.logits, init_model_outputs.logits))
+        self.assertTrue(torch.allclose(init_model_outputs.logits, transformers_outputs.logits, atol=1e-4))
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     def test_pipeline(self, model_arch):