Skip to content

Commit 4f99a34

Browse files
committed
Add fix to dockerfile for build
1 parent 5362e68 commit 4f99a34

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -209,6 +209,7 @@ ARG FLASH_ATT_VERSION=v2.3.6
209209

210210
WORKDIR /usr/src/flash-attention-v2
211211

212+
RUN pip install -U packaging --no-cache-dir
212213
# Download the wheel or build it if a pre-compiled release doesn't exist
213214
RUN MAX_JOBS=4 pip --verbose wheel flash-attn==${FLASH_ATT_VERSION} \
214215
--no-build-isolation --no-deps --no-cache-dir

0 commit comments

Comments
 (0)