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

axolotl-ai-cloud · Sep 19, 2023 · 97027e3 · 97027e3
1 parent 4629b90
commit 97027e3
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