tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix cd within flash-attn

Browse Source
This commit is contained in:
Wing Lian
2023-05-25 11:29:15 -04:00
parent 7ec105041d
commit f5fa3d131b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
docker/Dockerfile-base
View File

@@ -43,11 +43,11 @@ RUN git clone https://github.com/HazyResearch/flash-attention.git && \
python3 setup.py bdist_wheel && \ python3 setup.py bdist_wheel && \
cd csrc/fused_dense_lib && \ cd csrc/fused_dense_lib && \
python3 setup.py bdist_wheel && \ python3 setup.py bdist_wheel && \
cd csrc/xentropy && \ cd ../csrc/xentropy && \
python3 setup.py bdist_wheel && \ python3 setup.py bdist_wheel && \
cd csrc/rotary && \ cd ../csrc/rotary && \
python3 setup.py bdist_wheel && \ python3 setup.py bdist_wheel && \
cd csrc/layer_norm && \ cd ../csrc/layer_norm && \
python3 setup.py bdist_wheel python3 setup.py bdist_wheel
FROM base-builder AS deepspeed-builder FROM base-builder AS deepspeed-builder