commit change

mosaicml · May 9, 2024 · dc71f45 · dc71f45
1 parent 3a118d2
commit dc71f45
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -17,4 +17,4 @@ RUN git clone -b $BRANCH_NAME https://github.com/mosaicml/llm-foundry.git
 RUN pip install --no-cache-dir "./llm-foundry${DEP_GROUPS}"
 RUN pip uninstall -y llm-foundry
 RUN rm -rf llm-foundry
-RUN pip install --verbose git+https://github.com/NVIDIA/TransformerEngine.git@main
+RUN MAX_JOBS=1 pip install --verbose git+https://github.com/NVIDIA/TransformerEngine.git@main