Skip to content

Commit 486738e

Browse files
committed
build(torch): Downgrade flash-attn 3 to the 2.7.2.post1 tag
2.7.4.post1 has build conflicts flash-attn 2 and with compute_100
1 parent 77574c2 commit 486738e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torch/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ ARG BUILD_TORCH_VISION_VERSION="0.20.0"
77
ARG BUILD_TORCH_AUDIO_VERSION="2.5.0"
88
ARG BUILD_TRANSFORMERENGINE_VERSION="1.13"
99
ARG BUILD_FLASH_ATTN_VERSION="2.7.4.post1"
10-
ARG BUILD_FLASH_ATTN_3_VERSION="2.7.4.post1"
10+
ARG BUILD_FLASH_ATTN_3_VERSION="2.7.2.post1"
1111
ARG BUILD_TRITON_VERSION=""
1212
ARG BUILD_TRITON="1"
1313
ARG BUILD_TORCH_CUDA_ARCH_LIST="6.0 6.1 6.2 7.0 7.2 7.5 8.0 8.6 8.9 9.0+PTX"

0 commit comments

Comments
 (0)