fix device

Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
huggingface · Nov 26, 2024 · f0ba16a · f0ba16a
1 parent e7ec3d4
commit f0ba16a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/optimum/intel/ipex/modeling_base.py b/optimum/intel/ipex/modeling_base.py
@@ -308,8 +308,8 @@ def _init_warmup(self):
         if not self._add_patch:
             # use_cache = "past_key_values" in self.input_names
             dummy_inputs = _prepare_inputs_for_ipex_model(self, self.export_feature, self.use_cache)
-            if self._device.type != "cpu":
-                dummy_inputs = recursive_to_device(value=dummy_inputs, device=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)