ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Upgrade denpendency for xpu_lnl and xpu_arl option (#12424)

Browse source
This commit is contained in:
Yuwen Hu 2024-11-21 18:37:15 +08:00 committed by GitHub
parent 2935e97610
commit e61ae88c5b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/llm/setup.py
View file

@ -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,