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

[LLM] Add bigdl-core-xe dependency when installing bigdl-llm[xpu] (#8716)

Browse source 
* add bigdl-core-xe dependency
This commit is contained in:
xingyuan li 2023-08-10 17:41:42 +09:00 committed by GitHub
parent 7c56c39e36
commit 02ec01cb48
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/setup.py
View file

@ -253,7 +253,8 @@ def setup_package():
xpu_requires.remove('torch')
xpu_requires += ["torch==2.0.1a0",
"torchvision==0.15.2a0",
"intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'"]
"intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'",
"bigdl-core-xe;platform_system=='Linux'"]
metadata = dict(
name='bigdl-llm',