Skip to content

Commit 52445f4

Browse files
Update optimum/intel/openvino/modeling_seq2seq.py
Co-authored-by: Ella Charlaix <80481427+echarlaix@users.noreply.github.com>
1 parent 43ddf46 commit 52445f4

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/intel/openvino/modeling_seq2seq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -456,7 +456,7 @@ def _compile(self):
456456
if (
457457
"CACHE_DIR" not in ov_config.keys()
458458
and not str(self.parent_model.model_save_dir).startswith(gettempdir())
459-
and self._device.lower().split(":")[0] == "gpu"
459+
and "gpu" in self._device.lower()
460460
):
461461
cache_dir = Path(self.parent_model.model_save_dir).joinpath("model_cache")
462462
ov_config["CACHE_DIR"] = str(cache_dir)

0 commit comments

Comments
 (0)