update dockerfile to not build evoformer since it fails the build (#607)

This commit is contained in:
Wing Lian
2023-09-19 16:28:29 -04:00
committed by GitHub
parent 674c57692d
commit b53e77775b

View File

@@ -39,7 +39,7 @@ WORKDIR /workspace
RUN git clone https://github.com/microsoft/DeepSpeed.git && \
cd DeepSpeed && \
MAX_CONCURRENCY=8 DS_BUILD_SPARSE_ATTN=0 DS_BUILD_OPS=1 python3 setup.py bdist_wheel
MAX_CONCURRENCY=8 DS_BUILD_SPARSE_ATTN=0 DS_BUILD_OPS=1 DS_BUILD_EVOFORMER_ATTN=0 python3 setup.py bdist_wheel
FROM base-builder AS bnb-builder