support torch2.1 and ipex2.1 version

huggingface · Oct 23, 2023 · 9d03415 · 9d03415
1 parent 4011612
commit 9d03415
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,