Fix : VPTQ test (#35394)

fix_test
This commit is contained in:
Mohamed Mekkouri 2024-12-23 16:27:46 +01:00 committed by GitHub
parent 3a4ced9ab4
commit 59178780a6
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -44,7 +44,7 @@ class VptqConfigTest(unittest.TestCase):
quantization_config = VptqConfig()
vptq_orig_config = quantization_config.to_dict()
self.assertEqual(quantization_config.quant_config, vptq_orig_config["quant_config"])
self.assertEqual(vptq_orig_config["quant_method"], quantization_config.quant_method)
@slow