Dockerfile

FROM michaelf34/infinity:0.0.42

ENV HF_HOME=/tmp/cache
ENV INFINITY_QUEUE_SIZE=512
ENV INFINITY_LENGTHS_VIA_TOKENIZE="true"
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
ENV INFINITY_BATCH_SIZE=8
ENV INFINITY_PORT=7860

ENTRYPOINT ["infinity_emb", "v2"]