update dockerfile to not build evoformer since it fails the build (#607)
Browse files- docker/Dockerfile-base +1 -1
docker/Dockerfile-base
CHANGED
|
@@ -39,7 +39,7 @@ WORKDIR /workspace
|
|
| 39 |
|
| 40 |
RUN git clone https://github.com/microsoft/DeepSpeed.git && \
|
| 41 |
cd DeepSpeed && \
|
| 42 |
-
MAX_CONCURRENCY=8 DS_BUILD_SPARSE_ATTN=0 DS_BUILD_OPS=1 python3 setup.py bdist_wheel
|
| 43 |
|
| 44 |
FROM base-builder AS bnb-builder
|
| 45 |
|
|
|
|
| 39 |
|
| 40 |
RUN git clone https://github.com/microsoft/DeepSpeed.git && \
|
| 41 |
cd DeepSpeed && \
|
| 42 |
+
MAX_CONCURRENCY=8 DS_BUILD_SPARSE_ATTN=0 DS_BUILD_OPS=1 DS_BUILD_EVOFORMER_ATTN=0 python3 setup.py bdist_wheel
|
| 43 |
|
| 44 |
FROM base-builder AS bnb-builder
|
| 45 |
|