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
- Dockerfile +2 -1
- 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",
|