fix docker file

2024-06-27 20:29:16 +08:00 · 2024-06-27 20:29:16 +08:00 · 0f421055da
parent e44a4f07f0
commit 0f421055da
1 changed files with 2 additions and 2 deletions
--- a/docker/docker-cuda/Dockerfile
+++ b/docker/docker-cuda/Dockerfile
@ -26,8 +26,8 @@ RUN pip config set global.index-url "$PIP_INDEX" && \
 RUN pip uninstall -y transformer-engine flash-attn && \
    if [ "$INSTALL_FLASHATTN" == "true" ]; then \
        pip uninstall -y ninja && pip install ninja && \
-        pip install --no-cache-dir flash-attn --no-build-isolation \
-    fi;
+        pip install --no-cache-dir flash-attn --no-build-isolation; \
+    fi

 # Copy the rest of the application into the image
 COPY . /app