update dockerfile to not build evoformer since it fails the build (ax…

…olotl-ai-cloud#607)
mkeoliya · Sep 19, 2023 · 355fed6 · 355fed6
1 parent 203425d
commit 355fed6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile-base b/docker/Dockerfile-base
@@ -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