Skip to content

Commit 8cb7434

Browse files
committed
Fix bug (#4709)
1 parent f89f8c7 commit 8cb7434

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deploy/genai_vllm_server_docker/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ RUN python -m pip install "paddlex${PADDLEX_VERSION}"
1313

1414
ARG BUILD_FOR_SM120=false
1515
RUN if [ "${BUILD_FOR_SM120}" = 'true' ]; then \
16-
python -m pip install torch==2.8.0 https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.3.14/flash_attn-2.8.3+cu128torch2.8-cp310-cp310-linux_x86_64.whl; \
16+
python -m pip install torch==2.8.0 https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.4.11/flash_attn-2.8.3%2Bcu128torch2.8-cp310-cp310-linux_x86_64.whl; \
1717
else \
1818
python -m pip install torch==2.8.0 https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.3.14/flash_attn-2.8.2+cu128torch2.8-cp310-cp310-linux_x86_64.whl; \
1919
fi \

0 commit comments

Comments
 (0)