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
This commit is contained in:
Ren Xuancheng
2023-12-28 18:14:38 +08:00
committed by GitHub
2 changed files with 2 additions and 2 deletions

View File

@@ -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 \

View File

@@ -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 \