Spaces:
Running
on
A10G
Running
on
A10G
File size: 756 Bytes
9c1cd8c 8ed352c ef296f1 5101c69 ef296f1 072d5bd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
FROM ghcr.io/huggingface/text-embeddings-inference:1.2.2
RUN useradd -m -u 1000 user
# Switch to the "user" user
USER user
# Set home to the user's home directory
ENV HOME=/home/user \
PATH=/home/user/.local/bin:$PATH
# Set the working directory to the user's home directory
WORKDIR $HOME/app
# write secret to /home/user/.cache/huggingface/token
# first create the file and directory if needed
RUN mkdir -p /home/user/.cache/huggingface
RUN --mount=type=secret,id=HF_TOKEN,mode=0444,required=true echo $HF_TOKEN > /home/user/.cache/huggingface/token
ENTRYPOINT ["text-embeddings-router", "--model-id", "BAAI/bge-large-en-v1.5", "--hf-api-token", "$HF_TOKEN", "--api-key", "$HF_TOKEN", "--huggingface-hub-cache", "/home/user/.cache/huggingface"]
|