From d5038b716f4f5b2a31b0957f5a97e85bdc5a823d Mon Sep 17 00:00:00 2001
From: Chirag Jain <jain.chirag925@gmail.com>
Date: Mon, 4 Nov 2024 10:28:14 +0530
Subject: [PATCH] Update Dockerfile-notebook

---
 Dockerfile-notebook | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/Dockerfile-notebook b/Dockerfile-notebook
index 6e24567..7551b91 100644
--- 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 .