From f7a11d98f14796b9520cdb69edb90ffea39b1baa Mon Sep 17 00:00:00 2001 From: v-chen_data Date: Tue, 24 Sep 2024 15:49:25 -0700 Subject: [PATCH] small nit on comments --- release.Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/release.Dockerfile b/release.Dockerfile index a4a08e8b0e..0f397c23cd 100644 --- a/release.Dockerfile +++ b/release.Dockerfile @@ -17,6 +17,6 @@ RUN rm setup.py # Install TransformerEngine RUN NVTE_FRAMEWORK=pytorch CMAKE_BUILD_PARALLEL_LEVEL=4 MAX_JOBS=4 pip install git+https://github.com/NVIDIA/TransformerEngine.git@$TE_COMMIT -# Install and uninstall foundry to cache foundry requirements +# Install Foundry RUN git clone -b $BRANCH_NAME https://github.com/mosaicml/llm-foundry.git RUN pip install --no-cache-dir "./llm-foundry[all]"