wip

truefoundry · Mar 13, 2024 · 0bc7191 · 0bc7191
1 parent 27c6ebe
commit 0bc7191
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/build-and-push-image.yaml b/.github/workflows/build-and-push-image.yaml
@@ -93,7 +93,7 @@ jobs:
           cache-to: mode=max,image-manifest=true,type=registry,ref=${{ env.REPOSITORY_URI }}:buildcache
           build-args: |
             MAX_JOBS="1"
-            NVCC_APPEND_FLAGS="--verbose --threads 1"
+            NVCC_APPEND_FLAGS="--threads 1"
 
   # build_notebook_image:
   #   name: Build Notebook Image

diff --git a/Dockerfile b/Dockerfile
@@ -13,7 +13,7 @@ RUN mkdir -p /packages && \
     cd axolotl/ && \
     git checkout 43265208299242e3bc32690e22efadef79365c9d
 RUN cd /packages/axolotl/ && \
-    MAX_JOBS=${MAX_JOBS} NVCC_APPEND_FLAGS=${NVCC_APPEND_FLAGS} pip install -vvvvvvv -U --no-build-isolation -e .[deepspeed,flash-attn,mamba-ssm,fused-dense-lib] && \
+    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