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

Upgrade accelerate to 0.23.0 (#11331)

Browse source 
* Upgrade accelerate to 0.23.0
This commit is contained in:
Qiyuan Gong 2024-06-17 15:03:11 +08:00 committed by GitHub
parent 183e0c6cf5
commit 5d7c9bf901
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
python/llm/setup.py
View file

@ -54,8 +54,7 @@ libs_dir = os.path.join(llm_home, "ipex_llm", "libs")
cpu_torch_version = ["torch==2.1.2+cpu;platform_system=='Linux'", "torch==2.1.2;platform_system=='Windows'"] cpu_torch_version = ["torch==2.1.2+cpu;platform_system=='Linux'", "torch==2.1.2;platform_system=='Windows'"]
CONVERT_DEP = ['numpy == 1.26.4', # lastet 2.0.0b1 will cause error CONVERT_DEP = ['numpy == 1.26.4', # lastet 2.0.0b1 will cause error
'transformers == 4.36.2', 'sentencepiece', 'tokenizers == 0.15.2', 'transformers == 4.36.2', 'sentencepiece', 'tokenizers == 0.15.2',
# TODO: Support accelerate 0.22.0 'accelerate == 0.23.0', 'tabulate'] + cpu_torch_version
'accelerate == 0.21.0', 'tabulate'] + cpu_torch_version
SERVING_DEP = ['fschat[model_worker, webui] == 0.2.36', 'protobuf'] SERVING_DEP = ['fschat[model_worker, webui] == 0.2.36', 'protobuf']
windows_binarys = [ windows_binarys = [