Spaces:
Running
Running
Update Dockerfile
Browse files- Dockerfile +9 -2
Dockerfile
CHANGED
|
@@ -1,3 +1,10 @@
|
|
| 1 |
FROM michaelf34/infinity:0.0.41
|
| 2 |
-
|
| 3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
FROM michaelf34/infinity:0.0.41
|
| 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 |
+
ENV INFINITY_BATCH_SIZE=8
|
| 8 |
+
ENV INFINITY_PORT=7860
|
| 9 |
+
|
| 10 |
+
CMD ["infinity_emb", "v2"]
|