Skip to content

Commit

Permalink
Fix notebook build
Browse files Browse the repository at this point in the history
  • Loading branch information
chiragjn committed Dec 2, 2024
1 parent 0e49c66 commit 0ec15e0
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion Dockerfile-notebook
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@ RUN apt update && \

# Install torch and axolotl
USER jovyan
COPY requirements.txt notebook-requirements.txt /tmp/llm-finetune/
COPY 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
Expand Down

0 comments on commit 0ec15e0

Please sign in to comment.