Skip to content

Commit

Permalink
move PATH in Dockerfile (#857)
Browse files Browse the repository at this point in the history
fixes flash_attn installation
  • Loading branch information
pascal-pfeiffer authored Sep 13, 2024
1 parent 0568c22 commit 4bd792c
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -30,9 +30,7 @@ ENV HOME=/home/llmstudio
# Static application code lives in /workspace/
WORKDIR /workspace

# Add pip to the PATH
ENV PATH=/home/llmstudio/.local/bin:$PATH
ENV PATH=/workspace/.venv/bin:$PATH
RUN \
curl -sS https://bootstrap.pypa.io/get-pip.py | python3.10 && \
chmod -R a+w /home/llmstudio
Expand All @@ -43,6 +41,9 @@ COPY Makefile Pipfile Pipfile.lock /workspace/
ENV PIPENV_VENV_IN_PROJECT=1
RUN make setup && chmod -R 777 /workspace/.venv

# Add the venv to the PATH
ENV PATH=/workspace/.venv/bin:$PATH

# We need to create a mount point for the user to mount their volume
# All persistent data lives in /home/llmstudio/mount
RUN mkdir -p /home/llmstudio/mount
Expand Down

0 comments on commit 4bd792c

Please sign in to comment.