From 9d03415d46c7e7127395a1972056eb4733c56867 Mon Sep 17 00:00:00 2001 From: "Cheng, Penghui" 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..83185467b1 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<2.1.0", "onnx", "torch<2.1.0"], "diffusers": ["diffusers"], "quality": QUALITY_REQUIRE, "tests": TESTS_REQUIRE,