michaelfeil commited on
Commit
4a3e7c2
1 Parent(s): 715565a

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -3
Dockerfile CHANGED
@@ -1,10 +1,9 @@
1
  FROM michaelf34/infinity:0.0.42
2
 
3
  ENV HF_HOME=/tmp/cache
4
- ENV INFINITY_QUEUE_SIZE=512
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
- # wkcn/TinyCLIP-ViT-8M-16-Text-3M-YFCC15M
8
  ENV INFINITY_BATCH_SIZE=8
9
  ENV INFINITY_PORT=7860
10
 
 
1
  FROM michaelf34/infinity:0.0.42
2
 
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;wkcn/TinyCLIP-ViT-8M-16-Text-3M-YFCC15M;"
 
7
  ENV INFINITY_BATCH_SIZE=8
8
  ENV INFINITY_PORT=7860
9