Merge pull request #881 from QwenLM/fix-dockerfile-peft

Fix peft version in dockerfiles, causing a compatibility issue resulted from https://github.com/QwenLM/Qwen/pull/860
1 year ago · 4aab1d490b
parent 35023b6f2a 6c467b507b
commit 4aab1d490b
2 changed files with 2 additions and 2 deletions
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@ -67,7 +67,7 @@ if [ "$BUNDLE_FINETUNE" = "true" ]; then
    cd /data/shared/Qwen
    # Full-finetune / LoRA.
-    pip3 install deepspeed peft
+    pip3 install deepspeed "peft==0.5.0"
    # Q-LoRA.
    apt update -y && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \
--- a/docker/Dockerfile-cu114
+++ b/docker/Dockerfile-cu114
@ -63,7 +63,7 @@ if [ "$BUNDLE_FINETUNE" = "true" ]; then
    cd /data/shared/Qwen
    # Full-finetune / LoRA.
-    pip3 install deepspeed peft
+    pip3 install deepspeed "peft==0.5.0"
    # Q-LoRA.
    apt update -y && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \