feat: 只下载bge-small-zh-v1.5

Dockerfile

@@ -20,10 +20,11 @@ RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co
 RUN --mount=type=bind,from=infiniflow/ragflow_deps:latest,source=/huggingface.co,target=/huggingface.co \
     if [ "$LIGHTEN" != "1" ]; then \
         (tar -cf - \
-            /huggingface.co/BAAI/bge-large-zh-v1.5 \
-            /huggingface.co/BAAI/bge-reranker-v2-m3 \
-            /huggingface.co/maidalun1020/bce-embedding-base_v1 \
-            /huggingface.co/maidalun1020/bce-reranker-base_v1 \
+            /huggingface.co/BAAI/bge-small-zh-v1.5 \
+#            /huggingface.co/BAAI/bge-large-zh-v1.5 \
+#            /huggingface.co/BAAI/bge-reranker-v2-m3 \
+#            /huggingface.co/maidalun1020/bce-embedding-base_v1 \
+#            /huggingface.co/maidalun1020/bce-reranker-base_v1 \
             | tar -xf - --strip-components=2 -C /root/.ragflow) \
     fi
 

download_deps.py

@@ -23,10 +23,11 @@ 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",
-    "maidalun1020/bce-reranker-base_v1",
+    "BAAI/bge-small-zh-v1.5",
+#    "BAAI/bge-large-zh-v1.5",
+#    "BAAI/bge-reranker-v2-m3",
+#    "maidalun1020/bce-embedding-base_v1",
+#    "maidalun1020/bce-reranker-base_v1",
 ]
 
 def download_model(repo_id):