test

Dockerfile

@@ -7,7 +7,7 @@ FROM $BASE_IMAGE
 ARG BRANCH_NAME
 ARG DEP_GROUPS
 ARG TE_COMMIT
-ARG KEEP_FOUNDRY=false
+ARG KEEP_FOUNDRY="false"
 
 ENV TORCH_CUDA_ARCH_LIST="8.0 8.6 8.7 8.9 9.0"
 
@@ -20,9 +20,8 @@ RUN rm setup.py
 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
-WORKDIR /
 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 install --no-cache-dir "./llm-foundry${DEP_GROUPS}"
 
 # Conditionally uninstall llm-foundry and remove its directory
 RUN if [ "$KEEP_FOUNDRY" != "true" ]; then \