Style

Author: AlexKoff88

optimum/intel/__init__.py

@@ -63,7 +63,9 @@
         "OVWeightQuantizationConfig",
     ]
 else:
-    _import_structure["openvino"].extend(["OVConfig", "OVQuantizer", "OVTrainer", "OVTrainingArguments", "OVWeightQuantizationConfig"])
+    _import_structure["openvino"].extend(
+        ["OVConfig", "OVQuantizer", "OVTrainer", "OVTrainingArguments", "OVWeightQuantizationConfig"]
+    )
 
 try:
     if not (is_openvino_available() and is_diffusers_available()):
@@ -172,7 +174,13 @@
         if not (is_openvino_available() and is_nncf_available()):
             raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
-        from .utils.dummy_openvino_and_nncf_objects import OVConfig, OVQuantizer, OVTrainer, OVTrainingArguments, OVWeightQuantizationConfig
+        from .utils.dummy_openvino_and_nncf_objects import (
+            OVConfig,
+            OVQuantizer,
+            OVTrainer,
+            OVTrainingArguments,
+            OVWeightQuantizationConfig,
+        )
     else:
         from .openvino import OVConfig, OVQuantizer, OVTrainer, OVTrainingArguments, OVWeightQuantizationConfig
 

optimum/intel/openvino/quantization.py

@@ -279,16 +279,18 @@ def _quantize_ovcausallm(
         **kwargs,
     ):
         if self.model.stateful and not weights_only:
-            raise Exception("Full quantizaiton for stateful OVModelForCausalLM is currently broken. Possbile options:\n"
-                            "1. Quantize AutoModelForCausalLM\n"
-                            "2. Use weight only quantization\n"
-                            "3. Use stateful=False to export stateless model")
-        
+            raise Exception(
+                "Full quantizaiton for stateful OVModelForCausalLM is currently broken. Possbile options:\n"
+                "1. Quantize AutoModelForCausalLM\n"
+                "2. Use weight only quantization\n"
+                "3. Use stateful=False to export stateless model"
+            )
+
         save_directory = Path(save_directory)
         save_directory.mkdir(parents=True, exist_ok=True)
 
         if weights_only:
-            if quantization_config is None: 
+            if quantization_config is None:
                 # Use default 8-bit compression
                 self.model.model = nncf.compress_weights(self.model.model)
             else:

tests/openvino/test_quantization.py

@@ -66,7 +66,7 @@ class OVQuantizerTest(unittest.TestCase):
     # TODO : add models, enable OVModelForCausalLM.
     SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS = (
         (OVModelForSequenceClassification, "hf-internal-testing/tiny-random-bert", 32, 35),
-        #(OVModelForCausalLM, "hf-internal-testing/tiny-random-gpt2", 41, 23),
+        # (OVModelForCausalLM, "hf-internal-testing/tiny-random-gpt2", 41, 23),
     )
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES_WITH_EXPECTED_QUANTIZED_MATMULS)