This commit is contained in:
苏阳
2023-12-25 12:41:04 +08:00
committed by Ren Xuancheng
parent 0a5b3b6b9a
commit 65c73034c3
2 changed files with 4 additions and 4 deletions

View File

@@ -73,7 +73,7 @@ if [ "$BUNDLE_FINETUNE" = "true" ]; then
apt update -y && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \
libopenmpi-dev openmpi-bin \
&& rm -rf /var/lib/apt/lists/*
pip3 install optimum auto-gptq mpi4py
pip3 install "optimum==1.12.0" "auto-gptq==0.4.2" mpi4py
fi
EOF