huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 20:34:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: Fix tokenizer in llama.cpp Dockerfile

Browse Source
This commit is contained in:
Hugo Larcher 2024-11-29 00:25:17 +01:00
parent 26fca48a45
commit dfb4d5cd6a
No known key found for this signature in database
GPG Key ID: 3DAF63124699CA2B
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Dockerfile.llamacpp
View File

@ -66,11 +66,14 @@ RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
openssl \ openssl \
python3.11-dev \ python3.11-dev \
python3.11-venv \ python3.11-venv \
ibgomp1 libgomp1
COPY --from=builder /usr/src/text-generation-inference/target/release-opt/text-generation-backend-llamacpp /usr/src/text-generation-inference/text-generation-launcher COPY --from=builder /usr/src/text-generation-inference/target/release-opt/text-generation-backend-llamacpp /usr/src/text-generation-inference/text-generation-launcher
COPY --from=builder /usr/src/text-generation-inference/dist /usr/ COPY --from=builder /usr/src/text-generation-inference/dist /usr/
COPY --from=builder /usr/src/text-generation-inference/backends/llamacpp/requirements.txt requirements.txt
RUN /usr/bin/python3.11 -m venv /usr/src/text-generation-inference/venv
ENV PATH="/usr/src/text-generation-inference/venv/bin:$PATH"
RUN pip3 install --no-cache-dir -r requirements.txt
ENV PORT=8080 ENV PORT=8080
WORKDIR /usr/src/text-generation-inference WORKDIR /usr/src/text-generation-inference
ENTRYPOINT ["text-generation-launcher"] ENTRYPOINT ["text-generation-launcher"]