From 816c2cb1a6f58ca68d3eac737838f3fbb7187cde Mon Sep 17 00:00:00 2001
From: "Cheng, Penghui" <penghui.cheng@intel.com>
Date: Mon, 18 Dec 2023 16:40:59 +0800
Subject: [PATCH] Fixed UT error

Signed-off-by: Cheng, Penghui <penghui.cheng@intel.com>
---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 7946a0c8c5..00c1c63486 100644
--- a/setup.py
+++ b/setup.py
@@ -42,10 +42,11 @@
         "onnx",
         "onnxruntime<1.15.0",
         "transformers>=4.33.0",
+        "intel-extension-for-pytorch",
     ],
     "openvino": ["openvino>=2023.1.0", "onnx", "onnxruntime", "transformers>=4.33.0"],
     "nncf": ["nncf>=2.6.0"],
-    "ipex": ["transformers", "intel-extension-for-pytorch", "onnx"],
+    "ipex": ["transformers<4.32.0", "intel-extension-for-pytorch", "onnx"],
     "diffusers": ["diffusers"],
     "quality": QUALITY_REQUIRE,
     "tests": TESTS_REQUIRE,