Junjun Deng commited on
Commit
c44667c
·
1 Parent(s): 9ab6173

Add model id as env var

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