add flash-attn installation flag in Dockerfile

Former-commit-id: 2535044e95f6df628bd1f01e0eecb02407105d79
2024-06-27 00:11:04 +08:00
parent daea86e047
commit b46bd07119
4 changed files with 9 additions and 3 deletions
--- a/README_zh.md
+++ b/README_zh.md
@@ -444,6 +444,7 @@ docker build -f ./docker/docker-cuda/Dockerfile \
    --build-arg INSTALL_BNB=false \
    --build-arg INSTALL_VLLM=false \
    --build-arg INSTALL_DEEPSPEED=false \
+    --build-arg INSTALL_FLASH_ATTN=false \
    --build-arg PIP_INDEX=https://pypi.org/simple \
    -t llamafactory:latest .