From bfaf2447a82b8a8a612eb0fd1fb72779382e53f0 Mon Sep 17 00:00:00 2001 From: Chirag Jain Date: Mon, 2 Dec 2024 19:07:19 +0000 Subject: [PATCH] Fix build --- Dockerfile | 5 ++--- Dockerfile-notebook | 5 +++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/Dockerfile b/Dockerfile index 655bc87..b21da34 100644 --- a/Dockerfile +++ b/Dockerfile @@ -4,11 +4,10 @@ SHELL ["/bin/bash", "-c"] USER root # Install torch and axolotl requirements -COPY torch-requirements.txt base-requirements.txt requirements.txt /tmp/ +COPY torch-requirements.txt base-requirements.txt requirements.txt /tmp/llm-finetune/ RUN pip install -U pip wheel setuptools && \ pip uninstall -y axolotl && \ - pip install --no-cache-dir -U -r /tmp/torch-requirements.txt && \ - MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install -U --use-pep517 --no-build-isolation --no-cache-dir -r requirements.txt && \ + MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install -U --no-cache-dir --no-build-isolation --use-pep517 -r /tmp/llm-finetune/requirements.txt && \ rm -rf /root/.cache/pip # Install axolotl_truefoundry plugin diff --git a/Dockerfile-notebook b/Dockerfile-notebook index 4966afc..9d997f2 100644 --- a/Dockerfile-notebook +++ b/Dockerfile-notebook @@ -14,10 +14,11 @@ RUN apt update && \ # Install torch and axolotl USER jovyan -COPY base-requirements.txt requirements.txt notebook-requirements.txt /tmp/llm-finetune/ +COPY torch-requirements.txt base-requirements.txt requirements.txt notebook-requirements.txt /tmp/llm-finetune/ RUN pip install -U pip wheel setuptools && \ pip uninstall -y axolotl && \ - MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install --no-cache-dir -U --use-pep517 -r /tmp/llm-finetune/notebook-requirements.txt + pip install --no-cache-dir -U -r /tmp/llm-finetune/torch-requirements.txt && \ + MAX_JOBS=1 NVCC_APPEND_FLAGS="--threads 1" pip install -U --no-cache-dir --no-build-isolation --use-pep517 -r /tmp/llm-finetune/notebook-requirements.txt # Setup editable packages USER root