Skip to content

Commit 6289b57

Browse files
committed
change versions
1 parent 4897144 commit 6289b57

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

optimum/exporters/ipex/model_patcher.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -64,7 +64,7 @@ def patch_op(m, target_m, new_op_name, new_op):
6464

6565
def _patch_llama_model(model):
6666

67-
ipex_version = "2.2.0" if "xpu" in str(model.device) else "2.5.0"
67+
ipex_version = "2.1.0" if "xpu" in str(model.device) else "2.5.0"
6868
if is_ipex_version("<", ipex_version):
6969
raise ImportError(f"Only ipex version >= {ipex_version} supports RotaryEmbedding and IndirectAccessKVCache")
7070

optimum/intel/ipex/modeling_base.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@ def _is_patched_with_ipex(model, task):
7171
return True
7272
return False
7373
else:
74-
ipex_version = "2.2.0" if "xpu" in str(model.device) else "2.5.0"
74+
ipex_version = "2.1.0" if "xpu" in str(model.device) else "2.5.0"
7575
if is_ipex_version("<", ipex_version):
7676
return False
7777
return model.config.model_type in _IPEX_SUPPORT_MODEL_TYPES and task in _IPEX_EXPORTED_TASK

0 commit comments

Comments
 (0)