fix style

huggingface · Dec 11, 2023 · 39aab85 · 39aab85
1 parent 1fd9c58
commit 39aab85
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/optimum/gptq/quantizer.py b/optimum/gptq/quantizer.py
@@ -157,7 +157,7 @@ def __init__(
             "sym",
             "true_sequential",
             "quant_method",
-            "inside_layer_modules"
+            "inside_layer_modules",
         ]
 
         if self.bits not in [2, 3, 4, 8]:

diff --git a/tests/gptq/test_quantization.py b/tests/gptq/test_quantization.py
@@ -79,7 +79,7 @@ def setUpClass(cls):
             disable_exllama=cls.disable_exllama,
             exllama_config=cls.exllama_config,
             cache_block_outputs=cls.cache_block_outputs,
-            inside_layer_modules=cls.inside_layer_modules
+            inside_layer_modules=cls.inside_layer_modules,
         )
 
         cls.quantized_model = cls.quantizer.quantize_model(cls.model_fp16, cls.tokenizer)
@@ -302,18 +302,16 @@ class GPTQTestNoBlockCaching(GPTQTest):
     EXPECTED_OUTPUTS.add("Hello my name is jay and i am a student at university.")
     EXPECTED_OUTPUTS.add("Hello my name is John, I am a student in the University of")
 
+
 class GPTQTestInsideLayerModules(GPTQTest):
     # all layers are quantized apart from self_attention.dense
-    inside_layer_modules = [
-        ["self_attention.query_key_value"],
-        ["mlp.dense_h_to_4h"],
-        ["mlp.dense_4h_to_h"]
-    ]
+    inside_layer_modules = [["self_attention.query_key_value"], ["mlp.dense_h_to_4h"], ["mlp.dense_4h_to_h"]]
     EXPECTED_RELATIVE_DIFFERENCE = 1.57705236164535
 
     def test_not_converted_layers(self):
         # self_attention.dense should not be converted
-        self.assertTrue(self.quantized_model.transformer.h[0].self_attention.dense.__class__.__name__== "Linear")
+        self.assertTrue(self.quantized_model.transformer.h[0].self_attention.dense.__class__.__name__ == "Linear")
+
 
 class GPTQUtilsTest(unittest.TestCase):
     """