Skip to content

Commit

Permalink
ok
Browse files Browse the repository at this point in the history
  • Loading branch information
ef0xa committed Feb 10, 2025
1 parent 61c3ea4 commit b84e072
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -8,15 +8,15 @@ ARG WORKER_CUDA_VERSION=12.1.0

# Python dependencies.

RUN --mount=type=cache,target=/root/.cache/pip python3.11 -m pip install --upgrade pip
RUN --mount-type=cache,target=/root/.cache/pip python3.11 -m pip install --upgrade pip

# we're always going to do this. important to do this FIRST - it can take >2m and we want to cache the result as early as possible.
# TODO: pin this to a specific version

RUN --mount=type=cache,target=/root/.cache/pip python3.11 -m pip install torch torchvision torchaudio
RUN --mount-type=cache,target=/root/.cache/pip python3.11 -m pip install torch torchvision torchaudio

# ourother requirements may change; updating the version of infinity embedding, for instance.
COPY builder/requirements.txt /requirements.txt
RUN --mount=type=cache,target=/root/.cache/pip python3.11 -m pip install -r /requirements.txt
RUN --mount-type=cache,target=/root/.cache/pip python3.11 -m pip install -r /requirements.txt
COPY ./src /src
CMD python3.11 -u /src/handler.py

0 comments on commit b84e072

Please sign in to comment.