From 11fa5a8a0e38ad7a420c32be73607ec1b64af9a7 Mon Sep 17 00:00:00 2001 From: Heyang Sun <60865256+Uxito-Ada@users.noreply.github.com> Date: Thu, 23 Nov 2023 08:41:25 +0800 Subject: [PATCH] Fix QLoRA CPU dispatch_model issue about accelerate (#9506) --- python/llm/example/CPU/QLoRA-FineTuning/README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/python/llm/example/CPU/QLoRA-FineTuning/README.md b/python/llm/example/CPU/QLoRA-FineTuning/README.md index 3f4f555d..7f6a5e79 100644 --- a/python/llm/example/CPU/QLoRA-FineTuning/README.md +++ b/python/llm/example/CPU/QLoRA-FineTuning/README.md @@ -16,6 +16,7 @@ pip install --pre --upgrade bigdl-llm[all] pip install transformers==4.34.0 pip install peft==0.5.0 pip install datasets +pip install accelerate==0.23.0 ``` ### 2. Finetune model