Add version condition for IPEX prepare init (#1530)

changwangss · pre-commit-ci[bot] · web-flow · commit d96e14aff080 · 2024-01-11T09:22:55.000+08:00
* Update pytorch.py * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
diff --git a/neural_compressor/utils/pytorch.py b/neural_compressor/utils/pytorch.py
@@ -482,7 +482,10 @@ def recover_model_from_json(model, json_file_path, example_inputs):
     ipex = LazyImport("intel_extension_for_pytorch")
     from torch.ao.quantization.observer import MinMaxObserver
 
-    qconfig = ipex.quantization.get_smooth_quant_qconfig_mapping(alpha=0.5, act_observer=MinMaxObserver())
+    if ipex.__version__ >= "2.1.100":
+        qconfig = ipex.quantization.get_smooth_quant_qconfig_mapping(alpha=0.5, act_observer=MinMaxObserver)
+    else:
+        qconfig = ipex.quantization.get_smooth_quant_qconfig_mapping(alpha=0.5, act_observer=MinMaxObserver())
     if isinstance(example_inputs, dict):
         model = ipex.quantization.prepare(model, qconfig, example_kwarg_inputs=example_inputs, inplace=True)
     else:
