make env vars local

truefoundry · Mar 13, 2024 · 41b9c38 · 41b9c38
1 parent b9c108e
commit 41b9c38
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -2,8 +2,6 @@
 FROM --platform=linux/amd64 winglian/axolotl@sha256:ace38a300833e0e5fb22af7c8692306699ed3614a1f33b37e8bedb1f6ef0ef2e
 ARG MAX_JOBS
 ARG NVCC_APPEND_FLAGS
-ENV MAX_JOBS=${MAX_JOBS}
-ENV NVCC_APPEND_FLAGS=${NVCC_APPEND_FLAGS}
 USER root
 COPY requirements.txt /tmp/
 RUN pip install -U pip wheel setuptools && \
@@ -14,7 +12,7 @@ RUN mkdir -p /packages && \
     git clone https://github.com/OpenAccess-AI-Collective/axolotl && \
     cd axolotl/ && \
     git checkout 43265208299242e3bc32690e22efadef79365c9d
-RUN pip install -U --no-build-isolation -e .[deepspeed,flash-attn,mamba-ssm,fused-dense-lib] && \
+RUN MAX_JOBS=${MAX_JOBS} NVCC_APPEND_FLAGS=${NVCC_APPEND_FLAGS} pip install -U --no-build-isolation -e .[deepspeed,flash-attn,mamba-ssm,fused-dense-lib] && \
     pip uninstall -y mlflow tfy-mlflow-client && \
     pip install --no-cache-dir -U -r /tmp/requirements.txt && \
     rm -rf /root/.cache/pip