From e61ae88c5b816a03ff528b9a725283a01c2df7d6 Mon Sep 17 00:00:00 2001 From: Yuwen Hu <54161268+Oscilloscope98@users.noreply.github.com> Date: Thu, 21 Nov 2024 18:37:15 +0800 Subject: [PATCH] Upgrade denpendency for xpu_lnl and xpu_arl option (#12424) --- python/llm/setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/python/llm/setup.py b/python/llm/setup.py index 2e350623..90b9bbdd 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -297,9 +297,9 @@ def setup_package(): xpu_lnl_requires = copy.deepcopy(all_requires) for exclude_require in cpu_torch_version: xpu_lnl_requires.remove(exclude_require) - xpu_lnl_requires += ["torch==2.3.1+cxx11.abi", - "torchvision==0.18.1+cxx11.abi", - "intel-extension-for-pytorch==2.3.110+xpu", + xpu_lnl_requires += ["torch==2.3.1.post0+cxx11.abi;platform_system=='Windows'", + "torchvision==0.18.1.post0+cxx11.abi;platform_system=='Windows'", + "intel-extension-for-pytorch==2.3.110.post0+xpu;platform_system=='Windows'", "bigdl-core-xe-23==" + CORE_XE_VERSION, "bigdl-core-xe-batch-23==" + CORE_XE_VERSION, "bigdl-core-xe-addons-23==" + CORE_XE_VERSION,