Add HF transfer

Signed-off-by: Adrien Gallouët <angt@huggingface.co>
This commit is contained in:
Adrien Gallouët 2025-02-07 14:45:53 +00:00
parent 508d47f80f
commit 1401418243
No known key found for this signature in database
2 changed files with 3 additions and 0 deletions

View File

@ -71,4 +71,6 @@ COPY --from=builder /usr/lib/libllama.so /usr/lib/
COPY --from=builder /usr/lib/libggml*.so /usr/lib/
COPY --from=builder /app/target/release-opt/text-generation-router-llamacpp /usr/bin/
ENV HF_HUB_ENABLE_HF_TRANSFER=1
ENTRYPOINT ["text-generation-router-llamacpp"]

View File

@ -1,2 +1,3 @@
transformers==4.48.2
huggingface-hub==0.28.1
hf-transfer==0.1.9