Skip to content
Snippets Groups Projects
Unverified Commit 6a6ef12a authored by Yang Fan's avatar Yang Fan Committed by GitHub
Browse files

Update Dockerfile-cu121 

Fix <https://github.com/QwenLM/Qwen2/issues/512>.
parent 55bce75d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 2 additions and 2 deletions
docker/Dockerfile-cu121
+ 2
− 2
View file @ 6a6ef12a
...@@ -66,7 +66,7 @@ ARG BUNDLE_FLASH_ATTENTION=true ...@@ -66,7 +66,7 @@ ARG BUNDLE_FLASH_ATTENTION=true
RUN <<EOF RUN <<EOF
if [ "$BUNDLE_FLASH_ATTENTION" = "true" ]; then 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 fi
EOF EOF
...@@ -75,4 +75,4 @@ FROM bundle_flash_attention as final ...@@ -75,4 +75,4 @@ FROM bundle_flash_attention as final
COPY ../examples/sft/* ./ COPY ../examples/sft/* ./
COPY ../examples/demo/* ./ COPY ../examples/demo/* ./
EXPOSE 80 EXPOSE 80
\ No newline at end of file
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment