Update Dockerfile-notebook

truefoundry · Nov 4, 2024 · d5038b7 · d5038b7
1 parent 9f2b630
commit d5038b7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/Dockerfile-notebook b/Dockerfile-notebook
@@ -23,8 +23,7 @@ RUN cd /packages && \
     git checkout e16f637d079ef5d56321a240ef0547a50c37b708
 RUN cd /packages/axolotl/ && \
     MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install -U --no-build-isolation --no-cache-dir -e .[flash-attn,mamba-ssm,fused-dense-lib,optimizers,lion-pytorch,galore] && \
-    pip install --no-cache-dir -U -r /tmp/llm-finetune/notebook-requirements.txt && \
-    rm -rf /root/.cache/pip
+    pip install --no-cache-dir -U -r /tmp/llm-finetune/notebook-requirements.txt
 COPY plugins/axolotl_truefoundry /packages/axolotl_truefoundry
 RUN cd /packages/axolotl_truefoundry/ && \
     pip install --no-cache-dir -e .