fix(docker): fix docker image (#184)

This commit is contained in:
OlivierDehaene 2023-04-14 17:31:13 +02:00 committed by GitHub
parent 01c0e368e5
commit 1bb394631d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions

View File

@ -26,7 +26,7 @@ concurrency:
jobs:
build-and-push-image:
runs-on: ubuntu-latest
runs-on: large-runner
permissions:
contents: write
packages: write

View File

@ -132,6 +132,7 @@ WORKDIR /usr/src
RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
libssl-dev \
ca-certificates \
make \
&& rm -rf /var/lib/apt/lists/*
@ -148,7 +149,7 @@ COPY --from=transformers-builder /usr/src/transformers /usr/src/transformers
COPY --from=transformers-builder /usr/src/transformers/build/lib.linux-x86_64-cpython-39/transformers /usr/src/transformers/src/transformers
# Install transformers dependencies
RUN cd /usr/src/transformers && pip install -e . --no-cache-dir
RUN cd /usr/src/transformers && pip install -e . --no-cache-dir && pip install einops --no-cache-dir
# Install server
COPY proto proto