File size: 775 Bytes
0033842
9c1cd8c
01225c8
8ed352c
01225c8
 
8ed352c
01225c8
 
 
8ed352c
01225c8
 
8ed352c
ef296f1
5101c69
 
01225c8
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:86-1.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"]