Kevin Hu commited on
Commit
a891e52
·
1 Parent(s): c7d644d

Add huqie trie to docker image. (#4084)

Browse files

### What problem does this PR solve?



### Type of change

- [x] Performance Improvement

Files changed (2) hide show
  1. Dockerfile +2 -1
  2. download_deps.py +1 -0
Dockerfile CHANGED
@@ -12,10 +12,11 @@ WORKDIR /ragflow
12
  # Copy models downloaded via download_deps.py
13
  RUN mkdir -p /ragflow/rag/res/deepdoc /root/.ragflow
14
  RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co,target=/huggingface.co \
 
15
  tar --exclude='.*' -cf - \
16
  /huggingface.co/InfiniFlow/text_concat_xgb_v1.0 \
17
  /huggingface.co/InfiniFlow/deepdoc \
18
- | tar -xf - --strip-components=3 -C /ragflow/rag/res/deepdoc
19
  RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co,target=/huggingface.co \
20
  if [ "$LIGHTEN" != "1" ]; then \
21
  (tar -cf - \
 
12
  # Copy models downloaded via download_deps.py
13
  RUN mkdir -p /ragflow/rag/res/deepdoc /root/.ragflow
14
  RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co,target=/huggingface.co \
15
+ cp /huggingface.co/InfiniFlow/huqie/huqie.txt.trie /ragflow/rag/res/ && \
16
  tar --exclude='.*' -cf - \
17
  /huggingface.co/InfiniFlow/text_concat_xgb_v1.0 \
18
  /huggingface.co/InfiniFlow/deepdoc \
19
+ | tar -xf - --strip-components=3 -C /ragflow/rag/res/deepdoc
20
  RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co,target=/huggingface.co \
21
  if [ "$LIGHTEN" != "1" ]; then \
22
  (tar -cf - \
download_deps.py CHANGED
@@ -22,6 +22,7 @@ urls = [
22
  repos = [
23
  "InfiniFlow/text_concat_xgb_v1.0",
24
  "InfiniFlow/deepdoc",
 
25
  "BAAI/bge-large-zh-v1.5",
26
  "BAAI/bge-reranker-v2-m3",
27
  "maidalun1020/bce-embedding-base_v1",
 
22
  repos = [
23
  "InfiniFlow/text_concat_xgb_v1.0",
24
  "InfiniFlow/deepdoc",
25
+ "InfiniFlow/huqie",
26
  "BAAI/bge-large-zh-v1.5",
27
  "BAAI/bge-reranker-v2-m3",
28
  "maidalun1020/bce-embedding-base_v1",