Junjun Deng commited on
Commit
52e2399
·
1 Parent(s): c44667c

Rewrite the entrypoint

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -3,4 +3,5 @@ FROM ghcr.io/huggingface/text-embeddings-inference:cpu-1.2
3
  ARG MODEL_ID=Alibaba-NLP/gte-large-en-v1.5
4
  ENV MODEL_ID=${MODEL_ID}
5
 
6
- CMD ["--model-id", "${MODEL_ID}", "--port", "8080"]
 
 
3
  ARG MODEL_ID=Alibaba-NLP/gte-large-en-v1.5
4
  ENV MODEL_ID=${MODEL_ID}
5
 
6
+ ENTRYPOINT ["sh", "-c"]
7
+ CMD ["text-embeddings-router --model-id ${MODEL_ID} --port 8080"]