From 24b957242cebd9ee63f654a4877b1e2e3124d493 Mon Sep 17 00:00:00 2001
From: Chuck Tang <chuck.tang@databricks.com>
Date: Tue, 14 May 2024 11:48:38 -0700
Subject: [PATCH] commit change

---
 Dockerfile | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 316fc9bab0..2fa6b092d4 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -13,8 +13,7 @@ ADD https://raw.githubusercontent.com/mosaicml/llm-foundry/$BRANCH_NAME/setup.py
 RUN rm setup.py
 
 # Install and uninstall foundry to cache foundry requirements
-RUN git clone -b $BRANCH_NAME https://github.com/mosaicml/llm-foundry.git
-RUN pip install --no-cache-dir "./llm-foundry${DEP_GROUPS}"
-RUN export NVTE_FRAMEWORK=pytorch && MAX_JOBS=1 pip install --verbose git+https://github.com/NVIDIA/TransformerEngine.git@main
-RUN pip uninstall -y llm-foundry
-RUN rm -rf llm-foundry
+RUN git clone -b $BRANCH_NAME https://github.com/mosaicml/llm-foundry.git && \
+    pip install --no-cache-dir "./llm-foundry${DEP_GROUPS}" && \
+    pip uninstall -y llm-foundry && \
+    rm -rf llm-foundry && export NVTE_FRAMEWORK=pytorch && pip install --verbose git+https://github.com/NVIDIA/TransformerEngine.git@main