LLM: update setup to specify bigdl-core-xe version (#8913)

This commit is contained in:
Ruonan Wang 2023-09-07 15:11:55 +08:00 committed by GitHub
parent c152c719ea
commit c0797ea232

View file

@ -259,7 +259,7 @@ def setup_package():
xpu_requires += ["torch==2.0.1a0", xpu_requires += ["torch==2.0.1a0",
"torchvision==0.15.2a0", "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'"] "bigdl-core-xe==" + VERSION + ";platform_system=='Linux'"]
metadata = dict( metadata = dict(
name='bigdl-llm', name='bigdl-llm',