format

huggingface · Apr 16, 2024 · 79c5a16 · 79c5a16
1 parent 142a8cb
commit 79c5a16
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/neural_compressor/test_optimization.py b/tests/neural_compressor/test_optimization.py
@@ -520,7 +520,9 @@ def test_weight_only_quantization(self, methodology, weight_dtype):
         bits = 4 if "4" in weight_dtype else 8
         if methodology == "gptq":
             # max_input_length can be removed after neural-compressor > v2.5.1
-            quantization_config = GPTQConfig(bits=bits, sym=True, damp_percent=0.01, weight_dtype=weight_dtype, max_input_length=128)
+            quantization_config = GPTQConfig(
+                bits=bits, sym=True, damp_percent=0.01, weight_dtype=weight_dtype, max_input_length=128
+            )
         else:
             quantization_config = RtnConfig(bits=bits, weight_dtype=weight_dtype)