winglian commited on
Commit
8d6a289
1 Parent(s): 951facb

fix relative path in flash-attn build:

Browse files
Files changed (1) hide show
  1. docker/Dockerfile-base +3 -3
docker/Dockerfile-base CHANGED
@@ -43,11 +43,11 @@ RUN git clone https://github.com/HazyResearch/flash-attention.git && \
43
  python3 setup.py bdist_wheel && \
44
  cd csrc/fused_dense_lib && \
45
  python3 setup.py bdist_wheel && \
46
- cd ../csrc/xentropy && \
47
  python3 setup.py bdist_wheel && \
48
- cd ../csrc/rotary && \
49
  python3 setup.py bdist_wheel && \
50
- cd ../csrc/layer_norm && \
51
  python3 setup.py bdist_wheel
52
 
53
  FROM base-builder AS deepspeed-builder
 
43
  python3 setup.py bdist_wheel && \
44
  cd csrc/fused_dense_lib && \
45
  python3 setup.py bdist_wheel && \
46
+ cd ../xentropy && \
47
  python3 setup.py bdist_wheel && \
48
+ cd ../rotary && \
49
  python3 setup.py bdist_wheel && \
50
+ cd ../layer_norm && \
51
  python3 setup.py bdist_wheel
52
 
53
  FROM base-builder AS deepspeed-builder