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

feat: set transformers lib version. (#8683)

Browse source
This commit is contained in:
Cengguang Zhang 2023-08-04 15:01:59 +08:00 committed by GitHub
parent ef08250c21
commit ebcf75d506
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

@ -49,7 +49,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', 'sentencepiece', 'transformers >= 4.31.0', 'sentencepiece',
'accelerate', 'tabulate'] 'accelerate', 'tabulate']
windows_binarys = [ windows_binarys = [
"llama.dll", "llama.dll",