michaelfeil commited on
Commit
cb28108
·
verified ·
1 Parent(s): af5eee7

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -3,7 +3,9 @@ FROM michaelf34/infinity:0.0.46
3
  ENV HF_HOME=/tmp/cache
4
  ENV INFINITY_QUEUE_SIZE=256
5
  ENV INFINITY_LENGTHS_VIA_TOKENIZE="true"
6
- ENV INFINITY_MODEL_ID="michaelfeil/bge-small-en-v1.5;BAAI/bge-reranker-base;SamLowe/roberta-base-go_emotions;wkcn/TinyCLIP-ViT-8M-16-Text-3M-YFCC15M;"
 
 
7
  ENV INFINITY_BATCH_SIZE=8
8
  ENV INFINITY_PORT=7860
9
 
 
3
  ENV HF_HOME=/tmp/cache
4
  ENV INFINITY_QUEUE_SIZE=256
5
  ENV INFINITY_LENGTHS_VIA_TOKENIZE="true"
6
+ ENV INFINITY_MODEL_ID="jinaai/jina-clip-v1;michaelfeil/bge-small-en-v1.5;mixedbread-ai/mxbai-rerank-xsmall-v1;philschmid/tiny-bert-sst2-distilled;",
7
+ ENV INFINITY_REVISION="1cbe5e8b11ea3728df0b610d5453dfe739804aa9;ab7b31bd10f9bfbb915a28662ec4726b06c6552a;1d1a9dfbd0fde63df646402cf33e157e5852ead3;874eb28543ea7a7df80b6158bbf772d203efcab6;",
8
+
9
  ENV INFINITY_BATCH_SIZE=8
10
  ENV INFINITY_PORT=7860
11