Skip to content

D
dify
Unverified Commit 877da82b authored by takatost's avatar takatost Committed by GitHub
Browse files
Options

feat: cache huggingface gpt2 tokenizer files (#1138)

parent 66376290
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
api/Dockerfile
View file @ 877da82b
......@@ -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
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment