Merge pull request #660 from QwenLM/update-dockerfile-cu121

Update Dockerfile-cu121
QwenLM · Jun 17, 2024 · cc715a0 · cc715a0 · Domingo131352 · Jun 17, 2024
2 parents 55bce75 + 6a6ef12
commit cc715a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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
+EXPOSE 80