We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
flash-attn
1 parent 77574c2 commit 486738eCopy full SHA for 486738e
torch/Dockerfile
@@ -7,7 +7,7 @@ ARG BUILD_TORCH_VISION_VERSION="0.20.0"
7
ARG BUILD_TORCH_AUDIO_VERSION="2.5.0"
8
ARG BUILD_TRANSFORMERENGINE_VERSION="1.13"
9
ARG BUILD_FLASH_ATTN_VERSION="2.7.4.post1"
10
-ARG BUILD_FLASH_ATTN_3_VERSION="2.7.4.post1"
+ARG BUILD_FLASH_ATTN_3_VERSION="2.7.2.post1"
11
ARG BUILD_TRITON_VERSION=""
12
ARG BUILD_TRITON="1"
13
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