Update Embedding Mosec Dockerfile to use BAAI/bge-large-zh-v1.5 (#387)

This PR is used to update the Embedding Mosec dockerfile to use BAAI/bge-large-zh-v1.5. Signed-off-by: Tian, Feng <[email protected]>
opea-project · Aug 5, 2024 · bbdc1f0 · bbdc1f0
1 parent b2e2b1a
commit bbdc1f0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/comps/embeddings/langchain-mosec/mosec-docker/Dockerfile b/comps/embeddings/langchain-mosec/mosec-docker/Dockerfile
@@ -14,9 +14,9 @@ RUN pip3 install intel-extension-for-pytorch==2.2.0
 RUN pip3 install transformers
 RUN pip3 install llmspec mosec
 
-RUN cd /root/ && export HF_ENDPOINT=https://hf-mirror.com && huggingface-cli download --resume-download BAAI/bge-large-zh --local-dir /root/bge-large-zh
+RUN cd /root/ && export HF_ENDPOINT=https://hf-mirror.com && huggingface-cli download --resume-download BAAI/bge-large-zh-v1.5 --local-dir /root/bge-large-zh-v1.5
 
-ENV EMB_MODEL="/root/bge-large-zh/"
+ENV EMB_MODEL="/root/bge-large-zh-v1.5/"
 
 WORKDIR /root/comps/embeddings/langchain-mosec/mosec-docker