From bbdc1f048a2993c4446241a8672d814b95acc1ea Mon Sep 17 00:00:00 2001 From: "Tian, Feng" Date: Mon, 5 Aug 2024 14:09:11 +0800 Subject: [PATCH] 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 --- comps/embeddings/langchain-mosec/mosec-docker/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/comps/embeddings/langchain-mosec/mosec-docker/Dockerfile b/comps/embeddings/langchain-mosec/mosec-docker/Dockerfile index eb1e510a7..eec3020a4 100644 --- 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