From f00c442d404a6c6b71c0555bd4aa0f89f7f8fbe3 Mon Sep 17 00:00:00 2001
From: Zhao Changmin <changmin.zhao@intel.com>
Date: Tue, 12 Sep 2023 09:27:58 +0800
Subject: [PATCH] fix accelerate (#8946)

Co-authored-by: leonardozcm <leonardozcm@gmail.com>
---
 python/llm/setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/llm/setup.py b/python/llm/setup.py
index 756aa389..3af962b1 100644
--- 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",