Fix peft version in dockerfiles.

1 year ago · 6c467b507b
parent 35023b6f2a
commit 6c467b507b
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 \