From 6c467b507b70e978b8e80f82477c4e7c86e50f05 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=8B=8F=E9=98=B3?= Date: Thu, 28 Dec 2023 18:09:39 +0800 Subject: [PATCH] Fix peft version in dockerfiles. --- docker/Dockerfile | 2 +- docker/Dockerfile-cu114 | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 8dbc812..ac8337e 100644 --- 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 \ diff --git a/docker/Dockerfile-cu114 b/docker/Dockerfile-cu114 index d365554..25554d0 100644 --- 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 \