From 6a6ef12adb17ff102b2a6cc1df72fc6b95d53bcb Mon Sep 17 00:00:00 2001
From: Yang Fan <suyang.fy@alibaba-inc.com>
Date: Mon, 17 Jun 2024 15:38:47 +0800
Subject: [PATCH] Update Dockerfile-cu121

Fix <https://github.com/QwenLM/Qwen2/issues/512>.
---
 docker/Dockerfile-cu121 | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/docker/Dockerfile-cu121 b/docker/Dockerfile-cu121
index e1fc40d9..89d3b3f0 100644
--- a/docker/Dockerfile-cu121
+++ b/docker/Dockerfile-cu121
@@ -66,7 +66,7 @@ ARG BUNDLE_FLASH_ATTENTION=true
 
 RUN <<EOF 
 if [ "$BUNDLE_FLASH_ATTENTION" = "true" ]; then
-    pip3 install --no-cache-dir flash-attn==2.5.6 --no-build-isolation
+    pip3 install --no-cache-dir flash-attn==2.5.8 --no-build-isolation
 fi
 EOF
 
@@ -75,4 +75,4 @@ FROM bundle_flash_attention as final
 COPY ../examples/sft/* ./
 COPY ../examples/demo/* ./
 
-EXPOSE 80
\ No newline at end of file
+EXPOSE 80