update transformers version

huggingface · Jun 4, 2024 · f6b8010 · f6b8010
1 parent aaaa4c3
commit f6b8010
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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.39.0"):
+        if is_transformers_version("<", "4.38.2") or is_transformers_version(">=", "4.41.2"):
             raise ImportError("Only transformers versions 4.38.2 ~ 4.41.2 are verified.")
 
         from intel_extension_for_pytorch.llm.modules import Linear2SiluMul, LinearAdd

diff --git a/setup.py b/setup.py
@@ -62,7 +62,7 @@
     "neural-compressor": ["neural-compressor>=2.2.0", "onnxruntime<1.15.0", "accelerate"],
     "openvino": ["openvino>=2023.3", "nncf>=2.10.0", "openvino-tokenizers[transformers]"],
     "nncf": ["nncf>=2.10.0"],
-    "ipex": ["intel-extension-for-pytorch", "transformers>=4.36.0,<4.39.0"],
+    "ipex": ["intel-extension-for-pytorch", "transformers>=4.36.0,<=4.41.2"],
     "diffusers": ["diffusers"],
     "quality": QUALITY_REQUIRE,
     "tests": TESTS_REQUIRE,