diff --git a/python/llm/setup.py b/python/llm/setup.py index c6ae0581d9f..082492cc8ab 100644 --- a/python/llm/setup.py +++ b/python/llm/setup.py @@ -54,8 +54,7 @@ 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 'transformers == 4.36.2', 'sentencepiece', 'tokenizers == 0.15.2', - # TODO: Support accelerate 0.22.0 - 'accelerate == 0.21.0', 'tabulate'] + cpu_torch_version + 'accelerate == 0.23.0', 'tabulate'] + cpu_torch_version SERVING_DEP = ['fschat[model_worker, webui] == 0.2.36', 'protobuf'] windows_binarys = [