Fixed UT error

Signed-off-by: Cheng, Penghui <[email protected]>
huggingface · Dec 18, 2023 · 816c2cb · 816c2cb
1 parent b64eaa5
commit 816c2cb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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,