Apply suggestions from code review

no device_map

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

Author: faaany

optimum/intel/ipex/modeling_base.py

@@ -129,35 +129,13 @@ def __init__(
         **kwargs,
     ):
         OptimizedModel.__init__(self, model=model, config=config)
-        device_map = kwargs.pop("device_map", None)
         if device_map is None:
             if is_torch_xpu_available(check_device=True):
                 self._device = torch.device("xpu:0")
             elif torch.cuda.is_available():
                 self._device = torch.device("cuda:0")
             else:
                 self._device = torch.device("cpu")
-        else:
-            if isinstance(device_map, torch.device):
-                self._device = device_map
-            elif isinstance(device_map, str):
-                if device_map in ["auto", "balanced", "balanced_low_0", "sequential"]:
-                    raise ValueError(
-                        "When passing device_map as a string, the value needs to be a device name (e.g. cpu, xpu:0). "
-                        f"'auto', 'balanced', 'balanced_low_0', 'sequential' are not supported."
-                    )
-                self._device = torch.device(device_map)
-            elif isinstance(device_map, int):
-                if is_torch_xpu_available(check_device=True):
-                    self._device = torch.device(f"xpu:{device_map}")
-                elif torch.cuda.is_available():
-                    self._device = torch.device(f"cuda:{device_map}")
-                else:
-                    self._device = torch.device("cpu")
-            else:
-                raise ValueError(
-                    f"device_map should be either be a string, an integer or a torch.device object, but found {type(device_map)}"
-                )
         self.model.to(self._device)
         self._dtype = self.config.torch_dtype if self.config.torch_dtype is not None else torch.float32
         self.model_save_dir = model_save_dir