fromat

optimum/intel/neural_compressor/modeling_base.py

@@ -142,7 +142,9 @@ def _from_pretrained(
                 msg = f"Quantized model was obtained with torch version {inc_config.torch_version} but {_torch_version} was found."
                 logger.warning(f"{msg}")
         except EnvironmentError:
-            msg = f"Please check if torch quantization the model was obtained with is compatible with {_torch_version}."
+            msg = (
+                f"Please check if torch quantization the model was obtained with is compatible with {_torch_version}."
+            )
 
         if getattr(config, "backend", None) == "ipex" or getattr(config, "torchscript", False):
             # NOTE: Will improve to use load function when Intel Neural Compressor next 2.1 release.

optimum/intel/utils/import_utils.py

@@ -17,6 +17,7 @@
 import sys
 from collections import OrderedDict
 from typing import Union
+
 from packaging.version import Version, parse
 
 
@@ -205,7 +206,7 @@ def is_torch_version(operation: str, version: str):
     if not _torch_available:
         return False
 
-    import torch 
+    import torch
 
     return compare_versions(parse(parse(torch.__version__).base_version), operation, version)