File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed
deploy/genai_vllm_server_docker Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff line change @@ -13,7 +13,7 @@ RUN python -m pip install "paddlex${PADDLEX_VERSION}"
1313
1414ARG BUILD_FOR_SM120=false
1515RUN 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 \
You can’t perform that action at this time.
0 commit comments