fix accelerate (#8946)

Co-authored-by: leonardozcm <leonardozcm@gmail.com>
2023-09-12 09:27:58 +08:00 · 2023-09-12 09:27:58 +08:00 · f00c442d40
commit f00c442d40
parent 16761c58be
1 changed files with 2 additions and 1 deletions
--- a/python/llm/setup.py
+++ b/python/llm/setup.py
@ -51,7 +51,8 @@ github_artifact_dir = os.path.join(llm_home, '../llm-binary')
 libs_dir = os.path.join(llm_home, "bigdl", "llm", "libs")
 CONVERT_DEP = ['numpy >= 1.22', 'torch',
               'transformers == 4.31.0', 'sentencepiece',
-               'accelerate', 'tabulate']
+               # TODO: Support accelerate 0.22.0
+               'accelerate == 0.21.0', 'tabulate']
 windows_binarys = [
    "llama.dll",
    "gptneox.dll",