Update Dockerfile

neuralmagic · Mar 19, 2024 · 16c3f62 · 16c3f62
1 parent afb2b11
commit 16c3f62
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -115,6 +115,7 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 # Install flash attention (from pre-built wheel)
 RUN --mount=type=bind,from=flash-attn-builder,src=/usr/src/flash-attention-v2,target=/usr/src/flash-attention-v2 \
     pip install /usr/src/flash-attention-v2/*.whl --no-cache-dir
+
 #################### RUNTIME BASE IMAGE ####################