Lower flash-attn dependency version

huggingface · Dec 18, 2024 · 46a3be4 · 46a3be4
1 parent 4d1c1bb
commit 46a3be4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/containers/pytorch/training/gpu/2.3.0/transformers/4.47.0/py311/Dockerfile b/containers/pytorch/training/gpu/2.3.0/transformers/4.47.0/py311/Dockerfile
@@ -6,7 +6,7 @@ ARG DEBIAN_FRONTEND=noninteractive
 # Versions
 ARG CUDA="cu121"
 ARG PYTORCH="2.3.0"
-ARG FLASH_ATTN="2.7.2"
+ARG FLASH_ATTN="2.6.3"
 ARG TRANSFORMERS="4.47.0"
 ARG HUGGINGFACE_HUB="0.27.0"
 ARG DIFFUSERS="0.31.0"