Pārlūkot izejas kodu

feat: cache huggingface gpt2 tokenizer files (#1138)

tags/0.3.21
takatost pirms 2 gadiem
vecāks
revīzija
877da82b06
Revīzijas autora e-pasta adrese nav piesaistīta nevienam kontam
1 mainītis faili ar 3 papildinājumiem un 1 dzēšanām
  1. 3
    1
      api/Dockerfile

+ 3
- 1
api/Dockerfile Parādīt failu

@@ -26,7 +26,7 @@ EXPOSE 5001

WORKDIR /app/api

RUN apt-get update \
RUN apt-get update \
&& apt-get install -y --no-install-recommends bash curl wget vim nodejs \
&& apt-get autoremove \
&& rm -rf /var/lib/apt/lists/*
@@ -34,6 +34,8 @@ RUN apt-get update \
COPY --from=base /pkg /usr/local
COPY . /app/api/

RUN python -c "from transformers import GPT2TokenizerFast; GPT2TokenizerFast.from_pretrained('gpt2')"

COPY docker/entrypoint.sh /entrypoint.sh
RUN chmod +x /entrypoint.sh


Notiek ielāde…
Atcelt
Saglabāt