Hugging Face's logo

Spaces:
michaelfeil
/
infinity
Running

App Files Community
1
michaelfeil commited on
Commit
152934b
·
verified ·
1 Parent(s): d3d6dc1

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -0
Dockerfile CHANGED
@@ -3,6 +3,7 @@ FROM michaelf34/infinity:0.0.70-cpu
3
  # You could overwrite these defaults via Secrets
4
  ENV HF_HOME=/tmp/cache
5
  ENV INFINITY_QUEUE_SIZE=256
 
6
  ENV INFINITY_LENGTHS_VIA_TOKENIZE="true"
7
  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;"
8
  ENV INFINITY_REVISION="1cbe5e8b11ea3728df0b610d5453dfe739804aa9;ab7b31bd10f9bfbb915a28662ec4726b06c6552a;1d1a9dfbd0fde63df646402cf33e157e5852ead3;874eb28543ea7a7df80b6158bbf772d203efcab6;"
 
3
  # You could overwrite these defaults via Secrets
4
  ENV HF_HOME=/tmp/cache
5
  ENV INFINITY_QUEUE_SIZE=256
6
+ ENV INFINITY_ENGINE="torch"
7
  ENV INFINITY_LENGTHS_VIA_TOKENIZE="true"
8
  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;"
9
  ENV INFINITY_REVISION="1cbe5e8b11ea3728df0b610d5453dfe739804aa9;ab7b31bd10f9bfbb915a28662ec4726b06c6552a;1d1a9dfbd0fde63df646402cf33e157e5852ead3;874eb28543ea7a7df80b6158bbf772d203efcab6;"