diff --git a/.github/actions/llm/setup-llm-env/action.yml b/.github/actions/llm/setup-llm-env/action.yml index 1d530972..4b25ea0c 100644 --- a/.github/actions/llm/setup-llm-env/action.yml +++ b/.github/actions/llm/setup-llm-env/action.yml @@ -19,7 +19,7 @@ runs: sed -i 's/"bigdl-core-xe==" + CORE_XE_VERSION + "/"bigdl-core-xe/g' python/llm/setup.py sed -i 's/"bigdl-core-xe-esimd==" + CORE_XE_VERSION + "/"bigdl-core-xe-esimd/g' python/llm/setup.py sed -i 's/"bigdl-core-xe-21==" + CORE_XE_VERSION/"bigdl-core-xe-21"/g' python/llm/setup.py - sed -i 's/"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION + "/"bigdl-core-xe-esimd-21/g' python/llm/setup.py + sed -i 's/"bigdl-core-xe-esimd-21==" + CORE_XE_VERSION/"bigdl-core-xe-esimd-21"/g' python/llm/setup.py pip install requests if [[ ${{ runner.os }} == 'Linux' ]]; then diff --git a/python/llm/setup.py b/python/llm/setup.py index 02fabbbb..48689570 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -291,7 +291,7 @@ def setup_package(): "torchvision==0.16.0a0", "intel_extension_for_pytorch==2.1.10+xpu", "bigdl-core-xe-21==" + CORE_XE_VERSION, - "bigdl-core-xe-esimd-21==" + CORE_XE_VERSION + ";platform_system=='Linux'"] + "bigdl-core-xe-esimd-21==" + CORE_XE_VERSION] # default to ipex 2.1 for linux and windows xpu_requires = copy.deepcopy(xpu_21_requires)