Sarp Bilgiç commited on
Commit
8e22666
·
1 Parent(s): 8643dec

reranker warmup removed

Browse files
Files changed (2) hide show
  1. Dockerfile +6 -3
  2. requirements-prod.txt +1 -1
Dockerfile CHANGED
@@ -10,13 +10,16 @@ WORKDIR /app
10
  COPY requirements-prod.txt .
11
  RUN pip install --no-cache-dir -r requirements-prod.txt
12
 
 
 
 
13
  RUN python -c "\
14
  from fastembed import TextEmbedding, SparseTextEmbedding; \
15
- TextEmbedding(model_name='intfloat/multilingual-e5-large', cache_dir='./model_cache'); \
16
- SparseTextEmbedding(model_name='prithivida/Splade_PP_en_v1', cache_dir='./model_cache'); \
17
  "
18
 
19
- RUN python -c "from fastembed.rerank.cross_encoder import TextCrossEncoder; TextCrossEncoder(model_name='jinaai/jina-reranker-v1-turbo-en', cache_dir='./model_cache')"
20
 
21
  COPY . .
22
 
 
10
  COPY requirements-prod.txt .
11
  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')"
23
 
24
  COPY . .
25
 
requirements-prod.txt CHANGED
@@ -24,4 +24,4 @@ llama-index-tools-tavily-research
24
  qdrant-client
25
  tenacity
26
  llama-index-embeddings-fastembed
27
- fastembed
 
24
  qdrant-client
25
  tenacity
26
  llama-index-embeddings-fastembed
27
+ fastembed == 0.7.4