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

specify numpy version (#8489)

Browse source
This commit is contained in:
Yishuo Wang 2023-07-10 16:50:16 +08:00 committed by GitHub
parent 81d655cda9
commit 98bac815e4
1 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
python/llm/setup.py
View file

@ -45,7 +45,7 @@ BIGDL_PYTHON_HOME = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
VERSION = open(os.path.join(BIGDL_PYTHON_HOME, 'version.txt'), 'r').read().strip() VERSION = open(os.path.join(BIGDL_PYTHON_HOME, 'version.txt'), 'r').read().strip()
llm_home = os.path.join(os.path.dirname(os.path.abspath(__file__)), "src") llm_home = os.path.join(os.path.dirname(os.path.abspath(__file__)), "src")
libs_dir = os.path.join(llm_home, "bigdl", "llm", "libs") libs_dir = os.path.join(llm_home, "bigdl", "llm", "libs")
CONVERT_DEP = ['numpy', 'torch', 'transformers', 'sentencepiece', 'accelerate'] CONVERT_DEP = ['numpy >= 1.22', 'torch', 'transformers', 'sentencepiece', 'accelerate']
def get_llm_packages(): def get_llm_packages():
@ -73,7 +73,7 @@ def obtain_lib_urls():
"quantize-llama", "quantize-gptneox", "quantize-bloom", "quantize-llama", "quantize-gptneox", "quantize-bloom",
"main-llama_avx2", "main-gptneox_avx2", "main-bloom_avx2", "main-llama_avx2", "main-gptneox_avx2", "main-bloom_avx2",
"main-llama_avx512", "main-gptneox_avx512", "main-bloom_avx512", "main-llama_avx512", "main-gptneox_avx512", "main-bloom_avx512",
"libstarcoder_avx512.so", "main-starcoder_avx512", "quantize-starcoder", "libstarcoder_avx512.so", "main-starcoder_avx512", "quantize-starcoder",
"libstarcoder_avx2.so", "main-starcoder_avx2"] "libstarcoder_avx2.so", "main-starcoder_avx2"]
def get_date_urls(base_url): def get_date_urls(base_url):
@ -198,7 +198,7 @@ def setup_package():
for url in lib_urls[platform_name]: for url in lib_urls[platform_name]:
download_libs(url, change_permission=change_permission) download_libs(url, change_permission=change_permission)
all_requires = [] all_requires = []
all_requires += CONVERT_DEP all_requires += CONVERT_DEP