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

LLM: fix installation on windows (#8989)

Browse source
This commit is contained in:
binbin Deng 2023-09-18 11:14:54 +08:00 committed by GitHub
parent 52878d3e5f
commit 0a552d5bdc
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
python/llm/setup.py
View file

@ -50,7 +50,7 @@ llm_home = os.path.join(os.path.dirname(os.path.abspath(__file__)), "src")
github_artifact_dir = os.path.join(llm_home, '../llm-binary') github_artifact_dir = os.path.join(llm_home, '../llm-binary')
libs_dir = os.path.join(llm_home, "bigdl", "llm", "libs") libs_dir = os.path.join(llm_home, "bigdl", "llm", "libs")
CONVERT_DEP = ['numpy >= 1.22', 'torch', CONVERT_DEP = ['numpy >= 1.22', 'torch',
'transformers == 4.31.0', 'sentencepiece', 'transformers == 4.31.0', 'sentencepiece', 'tokenizers == 0.13.3',
# TODO: Support accelerate 0.22.0 # TODO: Support accelerate 0.22.0
'accelerate == 0.21.0', 'tabulate'] 'accelerate == 0.21.0', 'tabulate']
SERVING_DEP = ['fschat[model_worker, webui] == 0.2.28', 'protobuf'] SERVING_DEP = ['fschat[model_worker, webui] == 0.2.28', 'protobuf']