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 \