diff --git a/python/llm/setup.py b/python/llm/setup.py index c01f188c..5556f3d0 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -295,6 +295,15 @@ def setup_package(): # default to ipex 2.1 for linux and windows xpu_requires = copy.deepcopy(xpu_21_requires) + 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", + "bigdl-core-xe-23==" + CORE_XE_VERSION, + "bigdl-core-xe-batch-23==" + CORE_XE_VERSION, + "bigdl-core-xe-addons-23==" + CORE_XE_VERSION] cpp_requires = ["bigdl-core-cpp==" + CORE_XE_VERSION, "onednn-devel==2024.2.1;platform_system=='Windows'"] @@ -334,6 +343,7 @@ def setup_package(): "xpu": xpu_requires, # default to ipex 2.1 for linux and windows "npu": npu_requires, "xpu-2-1": xpu_21_requires, + "xpu-lnl": xpu_lnl_requires, "serving": serving_requires, "cpp": cpp_requires, "llama-index": llama_index_requires}, # for internal usage when upstreaming for llama-index