From c5f89bd58ce0aaab7f4d0fb95bd150fafffbacb5 Mon Sep 17 00:00:00 2001
From: "Cheng, Penghui" <penghui.cheng@intel.com>
Date: Mon, 23 Oct 2023 09:55:57 +0800
Subject: [PATCH] support torch2.1 and ipex2.1 version

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index d179b78a9d..777e41df45 100644
--- a/setup.py
+++ b/setup.py
@@ -46,7 +46,7 @@
     ],
     "openvino": ["openvino>=2023.1.0", "onnx", "onnxruntime"],
     "nncf": ["nncf>=2.6.0"],
-    "ipex": ["transformers<4.32.0", "intel-extension-for-pytorch", "onnx", "torch<2.1.0"],
+    "ipex": ["transformers<4.32.0", "intel-extension-for-pytorch", "onnx", "torch<=2.1.0"],
     "diffusers": ["diffusers"],
     "quality": QUALITY_REQUIRE,
     "tests": TESTS_REQUIRE,