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

fix accelerate (#8946)

Browse source 
Co-authored-by: leonardozcm <leonardozcm@gmail.com>
This commit is contained in:
Zhao Changmin 2023-09-12 09:27:58 +08:00 committed by GitHub
parent 16761c58be
commit f00c442d40
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/setup.py
View file

@ -51,7 +51,8 @@ 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',
'accelerate', 'tabulate'] # TODO: Support accelerate 0.22.0
'accelerate == 0.21.0', 'tabulate']
windows_binarys = [ windows_binarys = [
"llama.dll", "llama.dll",
"gptneox.dll", "gptneox.dll",