Spaces:

sofdog
/

live-transcription-docker

Build error

Sofia Casadei commited on 9 days ago

Commit

87d5645

1 Parent(s): 157b1a9

install flash attention

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -45,10 +45,9 @@ COPY --chown=user requirements.txt .
 RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
     /uv pip install -r requirements.txt
-# Try to install flash-attn, but don't fail the build if it errors
 RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
-    echo "Attempting to install flash-attn..." && \
-    (/uv pip install flash-attn --no-build-isolation || echo "Failed, continuing build...")
 # Copy application code
 COPY --chown=user . .

 RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
     /uv pip install -r requirements.txt
+# Install flash-attn from pre-built wheel
 RUN --mount=type=cache,target=$UV_CACHE_DIR,uid=1000,gid=1000 \
+    /uv pip install https://github.com/Dao-AILab/flash-attention/releases/download/v2.7.4.post1/flash_attn-2.7.4.post1+cu12torch2.6cxx11abiFALSE-cp311-cp311-linux_x86_64.whl
 # Copy application code
 COPY --chown=user . .