Update optimum/intel/neural_compressor/quantization.py

Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>

Author: PenghuiCheng

optimum/intel/neural_compressor/quantization.py

@@ -177,7 +177,7 @@ def quantize(
         save_directory.mkdir(parents=True, exist_ok=True)
         save_onnx_model = kwargs.pop("save_onnx_model", False)
         device = kwargs.pop("device", "cpu")
-        use_cpu = True if device == torch.device("cpu") or device == "cpu" else False
+        use_cpu = device == torch.device("cpu") or device == "cpu"
         use_xpu = device == torch.device("xpu") or device == "xpu"
 
         if save_onnx_model and (isinstance(self._original_model, ORTModel) or weight_only):