File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -44,7 +44,7 @@ COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/sympy-1.13.1.d
44
44
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/packaging /usr/local/lib/python3.12/dist-packages/packaging
45
45
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
46
46
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
48
48
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/
49
49
50
50
# 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
143
143
COPY --from=pytorch_image /usr/local/lib/python3.12/dist-packages/packaging /usr/local/lib/python3.12/dist-packages/packaging
144
144
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
145
145
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
147
147
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/
148
148
149
149
# Might not need to copy cusparseLt in the future once it's included in DLFW cuda container
You can’t perform that action at this time.
0 commit comments