From 742eef028fe3636337fb019c3701c635f46879c4 Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Wed, 18 Dec 2024 14:19:43 +0800 Subject: [PATCH] Add huqie trie to docker image. (#4084) ### What problem does this PR solve? ### Type of change - [x] Performance Improvement --- Dockerfile | 3 ++- download_deps.py | 1 + 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/Dockerfile b/Dockerfile index 56fb92f84..94365566c 100644 --- a/Dockerfile +++ b/Dockerfile @@ -12,10 +12,11 @@ WORKDIR /ragflow # Copy models downloaded via download_deps.py RUN mkdir -p /ragflow/rag/res/deepdoc /root/.ragflow RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co,target=/huggingface.co \ + cp /huggingface.co/InfiniFlow/huqie/huqie.txt.trie /ragflow/rag/res/ && \ tar --exclude='.*' -cf - \ /huggingface.co/InfiniFlow/text_concat_xgb_v1.0 \ /huggingface.co/InfiniFlow/deepdoc \ - | tar -xf - --strip-components=3 -C /ragflow/rag/res/deepdoc + | tar -xf - --strip-components=3 -C /ragflow/rag/res/deepdoc RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co,target=/huggingface.co \ if [ "$LIGHTEN" != "1" ]; then \ (tar -cf - \ diff --git a/download_deps.py b/download_deps.py index c859007c9..aa09c0ae5 100644 --- a/download_deps.py +++ b/download_deps.py @@ -22,6 +22,7 @@ urls = [ repos = [ "InfiniFlow/text_concat_xgb_v1.0", "InfiniFlow/deepdoc", + "InfiniFlow/huqie", "BAAI/bge-large-zh-v1.5", "BAAI/bge-reranker-v2-m3", "maidalun1020/bce-embedding-base_v1",