fix relative path in flash-attn build:
Browse files- 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 ../
|
47 |
python3 setup.py bdist_wheel && \
|
48 |
-
cd ../
|
49 |
python3 setup.py bdist_wheel && \
|
50 |
-
cd ../
|
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
|