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

Add setuptools limitation for ipex-llm[xpu] (#11102)

Browse source 
* Add setuptool limitation for ipex-llm[xpu]

* llamaindex option update
This commit is contained in:
Yuwen Hu 2024-05-22 18:20:30 +08:00 committed by GitHub
parent cd4dff09ee
commit d36b41d59e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/setup.py
View file

@ -281,6 +281,7 @@ def setup_package():
llama_index_requires = copy.deepcopy(all_requires) llama_index_requires = copy.deepcopy(all_requires)
for exclude_require in ['transformers == 4.31.0', 'tokenizers == 0.13.3'] + cpu_torch_version: for exclude_require in ['transformers == 4.31.0', 'tokenizers == 0.13.3'] + cpu_torch_version:
llama_index_requires.remove(exclude_require) llama_index_requires.remove(exclude_require)
llama_index_requires += ["setuptools<70.0.0"]
llama_index_requires += ["torch<2.2.0", llama_index_requires += ["torch<2.2.0",
"transformers>=4.34.0,<4.39.0", "transformers>=4.34.0,<4.39.0",
"sentence-transformers~=2.6.1"] "sentence-transformers~=2.6.1"]
@ -293,6 +294,7 @@ def setup_package():
xpu_21_requires = copy.deepcopy(all_requires) xpu_21_requires = copy.deepcopy(all_requires)
for exclude_require in cpu_torch_version: for exclude_require in cpu_torch_version:
xpu_21_requires.remove(exclude_require) xpu_21_requires.remove(exclude_require)
xpu_21_requires += ["setuptools<70.0.0"]
xpu_21_requires += ["torch==2.1.0a0", xpu_21_requires += ["torch==2.1.0a0",
"torchvision==0.16.0a0", "torchvision==0.16.0a0",
"intel_extension_for_pytorch==2.1.10+xpu", "intel_extension_for_pytorch==2.1.10+xpu",