From aaaa4c360e008a13c648537951db5928c5a2dbba Mon Sep 17 00:00:00 2001 From: jiqing-feng Date: Mon, 3 Jun 2024 12:31:46 -0400 Subject: [PATCH] fix transformers version --- optimum/exporters/ipex/modeling_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/optimum/exporters/ipex/modeling_utils.py b/optimum/exporters/ipex/modeling_utils.py index d6b79828d8..de41e914f3 100644 --- a/optimum/exporters/ipex/modeling_utils.py +++ b/optimum/exporters/ipex/modeling_utils.py @@ -227,7 +227,7 @@ class _IPEXLlamaDecoderLayerRef(nn.Module): def __init__(self, module, config, distributed=False): if is_ipex_version("<", "2.3.0"): raise ImportError("Only ipex version > 2.3.0 supports Linear2SiluMul and LinearAdd") - if is_transformers_version("<", "4.38.2") or is_transformers_version(">", "4.41.2"): + if is_transformers_version("<", "4.38.2") or is_transformers_version(">=", "4.39.0"): raise ImportError("Only transformers versions 4.38.2 ~ 4.41.2 are verified.") from intel_extension_for_pytorch.llm.modules import Linear2SiluMul, LinearAdd