tommulder commited on
Commit
77b66b9
·
1 Parent(s): 420a04f

chore(docker): try pip install flash-attn fallback when wheel tag mismatches

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -73,7 +73,8 @@ RUN if [ "$INSTALL_FLASH_ATTN" = "true" ]; then \
73
  echo "Installing flash-attn from $FLASH_ATTN_WHEEL_URL" && \
74
  pip install --no-cache-dir "$FLASH_ATTN_WHEEL_URL"; \
75
  else \
76
- echo "flash-attn wheel tag mismatch for $PYTAG. Skipping flash-attn install. Override FLASH_ATTN_WHEEL_URL if needed."; \
 
77
  fi; \
78
  else \
79
  echo "Skipping flash-attn installation"; \
 
73
  echo "Installing flash-attn from $FLASH_ATTN_WHEEL_URL" && \
74
  pip install --no-cache-dir "$FLASH_ATTN_WHEEL_URL"; \
75
  else \
76
+ echo "flash-attn wheel tag mismatch for $PYTAG. Trying pip install flash-attn (no-build-isolation)." && \
77
+ (pip install --no-cache-dir --no-build-isolation flash-attn || echo "flash-attn pip install failed; continuing without flash-attn"); \
78
  fi; \
79
  else \
80
  echo "Skipping flash-attn installation"; \