FROM michaelf34/infinity:0.0.51

ENV HF_HOME=/tmp/cache
ENV INFINITY_QUEUE_SIZE=256
ENV INFINITY_LENGTHS_VIA_TOKENIZE="true"
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;"
ENV INFINITY_REVISION="1cbe5e8b11ea3728df0b610d5453dfe739804aa9;ab7b31bd10f9bfbb915a28662ec4726b06c6552a;1d1a9dfbd0fde63df646402cf33e157e5852ead3;874eb28543ea7a7df80b6158bbf772d203efcab6;"
# ENV INFINITY_API_KEY=""
ENV INFINITY_BATCH_SIZE=8
ENV INFINITY_PORT=7860

ENTRYPOINT ["infinity_emb", "v2"]