starlight-apk/Qwen
1
0
Fork 0
mirror of https://github.com/QwenLM/Qwen.git synced 2026-05-20 16:35:47 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Dockerfile for <https://github.com/QwenLM/Qwen/issues/783>.

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
docker/Dockerfile
View File

@@ -73,7 +73,7 @@ if [ "$BUNDLE_FINETUNE" = "true" ]; then
apt update -y && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \ apt update -y && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \
libopenmpi-dev openmpi-bin \ libopenmpi-dev openmpi-bin \
&& rm -rf /var/lib/apt/lists/* && 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 fi
EOF EOF

6
docker/Dockerfile-cu114
View File

@@ -40,7 +40,7 @@ ARG BUNDLE_REQUIREMENTS=true
RUN <<EOF RUN <<EOF
if [ "$BUNDLE_REQUIREMENTS" = "true" ]; then if [ "$BUNDLE_REQUIREMENTS" = "true" ]; then
cd /data/shared/Qwen cd /data/shared/Qwen
pip3 install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113 pip3 install torch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2
pip3 install -r requirements.txt pip3 install -r requirements.txt
pip3 install -r requirements_web_demo.txt pip3 install -r requirements_web_demo.txt
fi fi
@@ -49,7 +49,7 @@ EOF
FROM bundle_req as bundle_flash_attention FROM bundle_req as bundle_flash_attention
ARG BUNDLE_FLASH_ATTENTION=true ARG BUNDLE_FLASH_ATTENTION=true
RUN <<EOF RUN <<EOF
if [ "$BUNDLE_FLASH_ATTENTION" = "true" ]; then if [ "$BUNDLE_FLASH_ATTENTION" = "true" ]; then
echo "CUDA 11.4 does not support flash-attention, please try other images." echo "CUDA 11.4 does not support flash-attention, please try other images."
fi fi
@@ -69,7 +69,7 @@ if [ "$BUNDLE_FINETUNE" = "true" ]; then
apt update -y && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \ apt update -y && DEBIAN_FRONTEND=noninteractive apt install -y --no-install-recommends \
libopenmpi-dev openmpi-bin \ libopenmpi-dev openmpi-bin \
&& rm -rf /var/lib/apt/lists/* && 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 fi
EOF EOF