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/docker/docker-cuda/docker-compose.yml
+++ b/docker/docker-cuda/docker-compose.yml
@@ -7,6 +7,7 @@ services:
        INSTALL_BNB: false
        INSTALL_VLLM: false
        INSTALL_DEEPSPEED: false
+        INSTALL_FLASH_ATTN: false
        PIP_INDEX: https://pypi.org/simple
    container_name: llamafactory
    volumes: