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

LLM: update setup.py to add bigdl-core-xe package (#9122)

Browse source
This commit is contained in:
Ruonan Wang 2023-10-10 15:02:48 +08:00 committed by GitHub
parent 1709beba5b
commit 388f688ef3
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

@ -261,7 +261,8 @@ def setup_package():
xpu_requires += ["torch==2.0.1a0",
"torchvision==0.15.2a0",
"intel_extension_for_pytorch==2.0.110+xpu;platform_system=='Linux'",
"bigdl-core-xe==" + VERSION + ";platform_system=='Linux'"]
"bigdl-core-xe==" + VERSION + ";platform_system=='Linux'",
"bigdl-core-xe-esimd==" + VERSION + ";platform_system=='Linux'"]
serving_requires = ['py-cpuinfo']
serving_requires += SERVING_DEP