Skip to content

Commit c1e1963

Browse files
authored
Update package versions in Dockerfile.triton.trt_llm_backend (#751)
1 parent 89ee0f2 commit c1e1963

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dockerfile/Dockerfile.triton.trt_llm_backend

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/sympy-1.13.1.d
4444
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/packaging /usr/local/lib/python3.12/dist-packages/packaging
4545
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/packaging-23.2.dist-info /usr/local/lib/python3.12/dist-packages/packaging-23.2.dist-info
4646
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn /usr/local/lib/python3.12/dist-packages/flash_attn
47-
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn-2.4.2.dev3.dist-info /usr/local/lib/python3.12/dist-packages/flash_attn-2.4.2.dev3.dist-info
47+
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn-2.7.3.dist-info /usr/local/lib/python3.12/dist-packages/flash_attn-2.7.3.dist-info
4848
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn_2_cuda.cpython-312-*-linux-gnu.so /usr/local/lib/python3.12/dist-packages/
4949

5050
# Might not need to copy cusparseLt in the future once it's included in DLFW cuda container
@@ -143,7 +143,7 @@ COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/sympy-1.13.1.d
143143
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/packaging /usr/local/lib/python3.12/dist-packages/packaging
144144
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/packaging-23.2.dist-info /usr/local/lib/python3.12/dist-packages/packaging-23.2.dist-info
145145
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn /usr/local/lib/python3.12/dist-packages/flash_attn
146-
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn-2.4.2.dev3.dist-info /usr/local/lib/python3.12/dist-packages/flash_attn-2.4.2.dev3.dist-info
146+
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn-2.7.3.dist-info /usr/local/lib/python3.12/dist-packages/flash_attn-2.7.3.dist-info
147147
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/flash_attn_2_cuda.cpython-312-*-linux-gnu.so /usr/local/lib/python3.12/dist-packages/
148148

149149
# Might not need to copy cusparseLt in the future once it's included in DLFW cuda container

0 commit comments

Comments
 (0)