diff --git a/tests/gptq/test_quantization.py b/tests/gptq/test_quantization.py index 20e2e3083c..dfba6e3465 100644 --- a/tests/gptq/test_quantization.py +++ b/tests/gptq/test_quantization.py @@ -113,8 +113,8 @@ def test_perplexity(self): the perplexity of the converted models """ - self.assertEqual(int(self.fp16_ppl), self.expected_fp16_perplexity) - self.assertEqual(int(self.quantized_ppl), self.expected_quantized_perplexity) + self.assertAlmostEqual(self.fp16_ppl, self.expected_fp16_perplexity, delta=1.0) + self.assertAlmostEqual(self.quantized_ppl, self.expected_quantized_perplexity, delta=1.0) def test_quantized_layers_class(self): """