Sarp Bilgiç commited on
Commit
dec5f60
·
1 Parent(s): b5533bb

fastembed default cache folder

Browse files
Files changed (1) hide show
  1. 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', cache_dir='/app/model_cache'); \
19
- SparseTextEmbedding(model_name='prithivida/Splade_PP_en_v1', cache_dir='/app/model_cache'); \
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')"