Apply suggestions from code review

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

Author: faaany

optimum/intel/ipex/modeling_base.py

@@ -324,7 +324,7 @@ def _init_warmup(self):
         if not self._is_ipex_exported:
             use_cache = "past_key_values" in self.input_names
             dummy_inputs = prepare_jit_inputs(self, self.export_feature, use_cache)
-            if "cpu" not in str(self._device):
+            if self._device.type != "cpu":
                 dummy_inputs = recursive_to_device(value=dummy_inputs, device=self._device)
             for _ in range(2):
                 self(**dummy_inputs)

optimum/intel/utils/modeling_utils.py

@@ -178,7 +178,7 @@ def recursive_to_device(value, device):
     if isinstance(value, (tuple, list)):
         return type(value)(recursive_to_device(v, device) for v in value)
     elif isinstance(value, dict):
-        return type(value)({k: recursive_to_device(v, device) for k, v in value.items()})
+        return {k: recursive_to_device(v, device) for k, v in value.items()}
     elif isinstance(value, torch.Tensor):
         return value.to(device)
     return value