Fix : VPTQ test (huggingface#35394)

fix_test
haotongl · Dec 23, 2024 · 4ceb30c · 4ceb30c
1 parent 2fa77b6
commit 4ceb30c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/quantization/vptq_integration/test_vptq.py b/tests/quantization/vptq_integration/test_vptq.py
@@ -44,7 +44,7 @@ def test_to_dict(self):
         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