We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 43ddf46 commit 52445f4Copy full SHA for 52445f4
optimum/intel/openvino/modeling_seq2seq.py
@@ -456,7 +456,7 @@ def _compile(self):
456
if (
457
"CACHE_DIR" not in ov_config.keys()
458
and not str(self.parent_model.model_save_dir).startswith(gettempdir())
459
- and self._device.lower().split(":")[0] == "gpu"
+ and "gpu" in self._device.lower()
460
):
461
cache_dir = Path(self.parent_model.model_save_dir).joinpath("model_cache")
462
ov_config["CACHE_DIR"] = str(cache_dir)
0 commit comments