From 02ec01cb48d7b237db6cdeabc194f997717c9fd8 Mon Sep 17 00:00:00 2001 From: xingyuan li <108672484+hoshibara@users.noreply.github.com> Date: Thu, 10 Aug 2023 17:41:42 +0900 Subject: [PATCH] [LLM] Add bigdl-core-xe dependency when installing bigdl-llm[xpu] (#8716) * add bigdl-core-xe dependency --- python/llm/setup.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/python/llm/setup.py b/python/llm/setup.py index be80ea23..8f2a3cd4 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -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',