Spaces:
Build error
Build error
Sarp Bilgiç commited on
Commit ·
dec5f60
1
Parent(s): b5533bb
fastembed default cache folder
Browse files- Dockerfile +3 -2
Dockerfile
CHANGED
|
@@ -12,11 +12,12 @@ RUN pip install --no-cache-dir -r requirements-prod.txt
|
|
| 12 |
|
| 13 |
ENV HF_HUB_DISABLE_SYMLINKS=1
|
| 14 |
ENV HF_HOME=/app/.cache/huggingface
|
|
|
|
| 15 |
|
| 16 |
RUN python -c "\
|
| 17 |
from fastembed import TextEmbedding, SparseTextEmbedding; \
|
| 18 |
-
TextEmbedding(model_name='intfloat/multilingual-e5-large'
|
| 19 |
-
SparseTextEmbedding(model_name='prithivida/Splade_PP_en_v1'
|
| 20 |
"
|
| 21 |
|
| 22 |
#RUN python -c "from fastembed.rerank.cross_encoder import TextCrossEncoder; TextCrossEncoder(model_name='jinaai/jina-reranker-v1-turbo-en', cache_dir='./model_cache')"
|
|
|
|
| 12 |
|
| 13 |
ENV HF_HUB_DISABLE_SYMLINKS=1
|
| 14 |
ENV HF_HOME=/app/.cache/huggingface
|
| 15 |
+
ENV FASTEMBED_CACHE_PATH=/app/model_cache
|
| 16 |
|
| 17 |
RUN python -c "\
|
| 18 |
from fastembed import TextEmbedding, SparseTextEmbedding; \
|
| 19 |
+
TextEmbedding(model_name='intfloat/multilingual-e5-large'); \
|
| 20 |
+
SparseTextEmbedding(model_name='prithivida/Splade_PP_en_v1'); \
|
| 21 |
"
|
| 22 |
|
| 23 |
#RUN python -c "from fastembed.rerank.cross_encoder import TextCrossEncoder; TextCrossEncoder(model_name='jinaai/jina-reranker-v1-turbo-en', cache_dir='./model_cache')"
|