diff --git a/python/llm/setup.py b/python/llm/setup.py index b6a30820..756aa389 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -259,7 +259,7 @@ 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;platform_system=='Linux'"] + "bigdl-core-xe==" + VERSION + ";platform_system=='Linux'"] metadata = dict( name='bigdl-llm',