ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Support llama-index install option for upstreaming purposes (#10866)

Browse source 
* Support llama-index install option for upstreaming purposes

* Small fix

* Small fix
This commit is contained in:
Yuwen Hu 2024-04-23 19:08:29 +08:00 committed by GitHub
parent 21bb8bd164
commit 5c9eb5d0f5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 11 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
python/llm/setup.py
View file

@ -275,6 +275,15 @@ def setup_package():
]
all_requires += CONVERT_DEP
# Add internal requires for llama-index
llama_index_requires = copy.deepcopy(all_requires)
for exclude_require in ['torch', 'transformers == 4.31.0', 'tokenizers == 0.13.3']:
llama_index_requires.remove(exclude_require)
llama_index_requires += ["torch<2.2.0",
"transformers>=4.34.0,<4.39.0",
"sentence-transformers~=2.6.1"]
# Linux install with --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
xpu_20_requires = copy.deepcopy(all_requires)
xpu_20_requires.remove('torch')
@ -324,7 +333,8 @@ def setup_package():
"xpu-2-0": xpu_20_requires,
"xpu-2-1": xpu_21_requires,
"serving": serving_requires,
"cpp": ["bigdl-core-cpp==" + CORE_XE_VERSION]},
"cpp": ["bigdl-core-cpp==" + CORE_XE_VERSION],
"llama-index": llama_index_requires}, # for internal usage when upstreaming for llama-index
classifiers=[
'License :: OSI Approved :: Apache Software License',
'Programming Language :: Python :: 3',